save predictions

[Niharikajo]

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