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

[KDP-wayofdata]

[zezhishao]

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

[KDP-wayofdata]

是的

[zezhishao]

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

[KDP-wayofdata]

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

[zezhishao]

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