Channels
random
linen
statsforecast
hierarchicalforecast
neural-forecast
mlforecast
general
timegpt
Powered by
#neural-forecast
Title
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!
Post