您好，请教一下模型训练好后如何获取预测值与真实值的数据 about basicts HOT 5 CLOSED

您是要获取测试集的预测值和真实值？

from basicts.

是的

from basicts.

可以参考这个test函数。
有其他疑问也欢迎继续讨论。

from basicts.

调用这个函数需要什么指令，可以直接输出为csv格式嘛

from basicts.

experiments/inference.py是一个推理脚本，加载训练好的 checkpoint 并在测试集上推理。
你在 test 函数结束之前加几行代码，把 test 函数里面的 returns_all存储到本地，然后运行这个脚本。
returns_all是一个字典，prediction 是你的预测、target 是真值，都是 torch.Tensor，具体如何保存、保存成什么格式可以自己选。

from basicts.