关于模型的保存与读取

liuxiyang641 / RAGAT

RAGAT: Relation Aware Graph Attention Network for Knowledge Graph Completion

56 stars 16 forks source link

关于模型的保存与读取 #9

Closed EurusNotes closed 2 years ago

EurusNotes commented 2 years ago

您好，不好意思，有几个问题想请教一下。想询问如何保存学习过后的模型，以及如何读取保存的模型。我试着跑了一下模型，MRR之类的结果是可以输出的。关于详细的预测结果的输出，这一部分应该是需要自己加的吧？

liuxiyang641 commented 2 years ago

你好，模型本身的参数在训练过程中达到最好结果的时候会自动保存在checkpoints/文件夹下，如果后续使用可以通过命令行参数-restore和-name checkpoints下的对应文件名来重新加载特定的模型参数；如果希望更改输出的评估指标的结果，可以自己通过修改run.py文件中的evaluate()函数进行变动。

EurusNotes commented 2 years ago

谢谢您的回复！