https://github.com/nixtla logo
k

Kevin

05/09/2023, 3:58 PM
I am trying to use Temporal Fusion Transformer (TFT) (https://nixtla.github.io/neuralforecast/models.tft.html#tft). Can I extract variable importances from NeuralForecast's implementation of TFT? I am referring to Pytorch's implementation of TFT (https://pytorch-forecasting.readthedocs.io/en/stable/tutorials/stallion.html). Thanks.
c

Cristian (Nixtla)

05/09/2023, 4:05 PM
Hi @Kevin. Our implementation also allows for extracting variable importance. We dont have an equivalent
interpret_output
method yet. It is currently in our backlog and we will implement it soon!