get_stallion_data

pytorch_forecasting.data.examples.get_stallion_data() pandas.core.frame.DataFrame[source]

Demand data with covariates.

~20k samples of 350 timeseries. Important columns

  • Timeseries can be identified by agency and sku.

  • volume is the demand

  • date is the month of the demand.

Returns

data

Return type

pd.DataFrame