search

thuml / Autoformer

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008
MIT License
2k stars 429 forks source link

训练自己的数据集 #139

Closed Dyw08 closed 1 year ago

Dyw08 commented 1 year ago

你好，我自己的数据集只有三类，但ETTh1的数据集有7类，我运行显示通道不对，这个要怎么处理？

wuhaixu2016 commented 1 year ago

您好，请修改enc_in，c_out参数

Dyw08 commented 1 year ago

好的，可以成功训练了!非常感谢！！