-
Notifications
You must be signed in to change notification settings - Fork 96
Open
Description
陈老师,您好^-^!我想问一下关于模型保存、加载做预测的问题。因为不想每次都迭代训练模型,所以我想把模型图及其参数都保存下来,使用的语句如下:
这个语句将模型训练过程中的所有参数以及图模型都保存下来了:
然后利用saver.restore()加载模型参数,仍然对测试集进行预测。在训练模型的过程中,对测试集进行测试准确率计算都会达到90%左右的分数。 但是通过加载保存模型参数的方式,仍对测试集进行准确率计算,不仅达不到90%,而且每次预测都会得到的准确率都是随机的。
以上是自己写的预测函数。感觉自己加载了原始的图模型,也restore了模型的参数。然后就是准备预测的数据集,计算准确率。 不知道为什么,总是得到一个分数很低,且又很随机的准确率分数???不能和trian模型是,预测训练集得到的准确率分数保持一致。
其实,我还尝试了自己建一个简单的模型,数据集仍然用您的。同样可以在训练过程中得到好的准确率分数,测试集也能得到一个好的分数。但是通过加载保存的模型图和模型参数,又对测试集进行预测,仍然得到一个随机的,且分数很低的准确率分数。
陈老师,您有空的时候,帮忙解答一下。很想解决这个问题。期待您的回复^-^
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels






