lucidrains / iTransformer

Unofficial implementation of iTransformer - SOTA Time Series Forecasting using Attention networks, out of Tsinghua / Ant group
MIT License
445 stars 36 forks source link

Help to use the code #29

Open zhouzhq2021 opened 3 months ago

zhouzhq2021 commented 3 months ago

How to use custom dataset to train own model or that's just a architecture? Anybody try to do some experiences in your own dataset?

Dante-Berth commented 3 months ago

Yeah it is just an architecture. You can use Pytorch Lighting for instance to train your model which can be a composition of iTransformer and a simple Linear.