timeseriesAI / tsai

Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai
https://timeseriesai.github.io/tsai/
Apache License 2.0
5.14k stars 642 forks source link

Using MVP callback for pre-training a TSTabFusionTransformer multilabel classification model #865

Open a-millarch opened 10 months ago

a-millarch commented 10 months ago

First of all, thank you for this inspiring and useful library.

I wish to pre-train a TSTabFusionTransformer model and I found the MVP callback (and tutorial). It seems, I would need to allow a custom head for the model class and also modify the MVP callback to handle the input shape and further how the MVP custom head is created.

Do you have any suggestions on how to achieve this, if even possible?