模型后加载模型,精度变差"/>
pygcn保存模型后加载模型,精度变差
最近在看pygcn的代码,用pytorch实现的GCN。
但是在保存模型以后,重新加载模型,测试,发现精度大大降低。经过一下午的查找,发现是在进行独热编码的时候,classes=set(labels)
,这使得每次的classes中标签的顺序都不一样。
比如上图中是一次编码的顺序,但是由于set
中的元素是无序的,下一次可能Theory就是0了。
#classes = set(labels) # 不按顺序classes = list(set(labels))classes.sort()
改成列表就好了。
更多推荐
pygcn保存模型后加载模型,精度变差
发布评论