cure-lab / SCINet

The GitHub repository for the paper: “Time Series is a Special Sequence: Forecasting with Sample Convolution and Interaction“. (NeurIPS 2022)
Apache License 2.0
617 stars 127 forks source link

如何训练自己的数据集呢? #43

Closed DomineeringDragon closed 2 years ago

DomineeringDragon commented 2 years ago

在下菜鸟一枚。之前在informer模型上运行了自己的数据集,现在打算在SCINet上也如法炮制一番。但是遇到了如下的报错: Q_O7} ~}GKAMSQ@RUO)62_Y 我是这样处理的:将run_ETTh.py文件中的data_paser添加我的数据集,对应的dataloader是Dataset_Custom,并在ETTH_data_loader.py文件中添加了配置。加载应该是可以的,但是返回了报错。Etth数据集加载的代码结构和informer模型代码的很像,但在informer模型的代码中就没有报错。请问是为什么呢?

bendoesai commented 2 years ago

exp_ETTh.py image

line 28: 7 -> 129

I had a similar error. I found that while using a custom dataset, I needed to manually change the number of feature columns in exp_ETTh.py to the correct number for my dataset.

This should fix your issue.

VEWOXIC commented 2 years ago

Yes, we recommend to use the ETT dataloader for training your own dataset. All you need is to change the input dimension as your dataset. The ETT dataloader support both the univariant and multi-variant task.

VEWOXIC commented 2 years ago

If there is no further problem, We will close this issue now.