Using Custom Dataset with Different Input Lengths for Model

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

Using Custom Dataset with Different Input Lengths for Model #220

Closed rozhix closed 1 month ago

rozhix commented 4 months ago

My input dimensions are [1500, 50, 6] and my label dimensions are [1500, 1]. Can I use this model? If so, could you please explain how to configure it correctly?

wuhaixu2016 commented 1 month ago

Sorry for the delayed response. I am not sure if I have correctly understood your question, I think it may be more like a classification task. You can see the dataloader of the classification task in TSlib (https://github.com/thuml/Time-Series-Library/blob/main/data_provider/data_loader.py#L616)