search

zalandoresearch / pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend
MIT License
1.21k stars 190 forks source link

how to customize dataset? #31

Closed bugsuse closed 3 years ago

bugsuse commented 3 years ago

It's a great work! And I want to apply the model to predict air quality such as PM10 and PM2.5 based on temperature, wind speed and direction and so on. Can this model directly accept pytorch's dataloader object? If not, how to customize dataset?

kashif commented 3 years ago

@bugsuse thanks! So you need to convert your data to the actual format the models accept. Have a look at the _m5 data set file where I prepare such a dataset. Once you have a dataset we do use PyTorch DataLoaders to make batches etc.

bugsuse commented 3 years ago

@kashif Thanks so much! I have solved it and closed it.