Prediction on entire dataset

thuml / Koopa

Code release for "Koopa: Learning Non-stationary Time Series Dynamics with Koopman Predictors" (NeurIPS 2023), https://arxiv.org/abs/2305.18803

MIT License

186 stars 23 forks source link

Prediction on entire dataset #9

Closed FilmonYacob closed 11 months ago

FilmonYacob commented 11 months ago

Hello,

Thank you for this project and contribution.

Currently, the model predicts only one batch given a CSV file. Is there a convinient way to modify it so that it can predict on the entire dataset, similar to how predictions are made on the test set?

WenWeiTHU commented 11 months ago

Hello, you can refer to the testfunction in exp_main.py