get_stallion_data

pytorch_forecasting.data.examples.get_stallion_data() → DataFrame

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