DAMO-DI-ML / ICML2022-FEDformer

Source code of ICML'22 paper: FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecasting
MIT License
197 stars 34 forks source link

请问如何使用自己的数据集? #2

Closed David-Ttao closed 2 years ago

David-Ttao commented 2 years ago

我想把fedformer用于自己构造的数据集上,请问替换路径后还需要修改什么部分?

tianzhou2011 commented 2 years ago

如果是在我们框架下跑的话,你要看看data_provider/data_loader.py 写一个自己的dataloader. 但一般来说你会有自己的训练和数据处理框架吧?你把fedformer model文件挪过去跑就好了,它就是个seq2seq的模型

David-Ttao commented 2 years ago

谢谢解答!