search

kashif / pytorch-transformer-ts

Repository of Transformer based PyTorch Time Series Models
MIT License
283 stars 38 forks source link

transformer/multivariate.ipynb GluonTSDataError #15

Open kevindarby opened 1 year ago

kevindarby commented 1 year ago

File ~/code/cqg/bts/spark/ml/.venv/lib/python3.9/site-packages/gluonts/torch/model/estimator.py:205, in PyTorchLightningEstimator.train_model(self, training_data, validation_data, from_predictor, shuffle_buffer_length, cache_data, ckpt_path, *kwargs) ... 93 """ 94 if not condition: ---> 95 raise exception_class(message.format(args, **kwargs))

GluonTSDataError: Input for field "target" does not have the requireddimension (field: target, ndim observed: 2, expected ndim: 1)

kevindarby commented 1 year ago

gluonts==0.13.1

kevindarby commented 1 year ago

happens on the train call

predictor = estimator.train( training_data=dataset_train, num_workers=8, shuffle_buffer_length=1024 )

kashif commented 1 year ago

hmm i see let me confirm... i might need to update it for the new gluonts api

kevindarby commented 1 year ago

Thanks StudentTOutput changed too

kashif commented 1 year ago

indeed