oliverguhr/transformer-time-series-prediction

How to predict in multi-dimension

xiaoerlaigeid opened this issue · 1 comments

xiaoerlaigeid commented

Hello I want to know in this code the prediction is one dimension. But in my problem, the prediction is three-dimensions. I want to know how to modify this code.

oliverguhr commented

You just need to modify this line:

transformer-time-series-prediction/transformer-singlestep.py

Line 51 in 570d39b

self.decoder = nn.Linear(feature_size,1)

It reduces the models feature to one dimension, simply put in a two and you get two output dimensions.

  • 👍1