BasicTS icon indicating copy to clipboard operation
BasicTS copied to clipboard

如何保存预测结果

Open lingy-qd opened this issue 1 year ago • 1 comments

您好,请问如何将预测结果和真实值保存在npy文件中?我不太熟悉easytorch框架,请提供给我一些信息,谢谢!

lingy-qd avatar Nov 27 '23 12:11 lingy-qd

You can save the prediction values of the test dataset by modifying test function in runners.base_tsf_runner.py. You can save the prediction and real_value before this line.

zezhishao avatar Nov 30 '23 03:11 zezhishao

您好,目前BasicTS已经可以自动保存预测结果,每次训练完成后,BasicTS使用最好的checkpoint进行推理,推理结果保存在test_results.npz中。

zezhishao avatar Sep 28 '24 09:09 zezhishao