moraieu / query-selector

LONG-TERM SERIES FORECASTING WITH QUERYSELECTOR – EFFICIENT MODEL OF SPARSEATTENTION
Apache License 2.0
75 stars 18 forks source link

save predictions #6

Open Niharikajo opened 2 years ago

Niharikajo commented 2 years ago

Hello Jacek,

I was going through the query selector paper, and I am trying to reproduce your code. I have a few doubts :

  1. Where is the train, test, and validation split on the dataset done?

  2. In train.py we train the model, and then validation is done. Where is testing happening?

  3. If our pred length is 24, we get 24 predictions after the last data point, So how can we save the values predicted by the model? In which part of the code is happening ?

Please do let me know

Thank you Regards Niharika Joshi