mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-08-12 12:20:53 +08:00
10 lines
244 B
Python
10 lines
244 B
Python
from functools import partial
|
|
|
|
from gluonts.dataset.repository.datasets import dataset_recipes
|
|
|
|
from ._m5 import generate_pts_m5_dataset
|
|
|
|
dataset_recipes["pts_m5"] = partial(
|
|
generate_pts_m5_dataset, pandas_freq="D", prediction_length=28
|
|
)
|