pygcn保存模型后加载模型,精度变差

编程入门 行业动态 更新时间:2024-10-09 00:40:54

pygcn保存<a href=https://www.elefans.com/category/jswz/34/1771358.html style=模型后加载模型,精度变差"/>

pygcn保存模型后加载模型,精度变差

最近在看pygcn的代码,用pytorch实现的GCN。
但是在保存模型以后,重新加载模型,测试,发现精度大大降低。经过一下午的查找,发现是在进行独热编码的时候,classes=set(labels),这使得每次的classes中标签的顺序都不一样。

比如上图中是一次编码的顺序,但是由于set中的元素是无序的,下一次可能Theory就是0了。

    #classes = set(labels) # 不按顺序classes = list(set(labels))classes.sort()

改成列表就好了。

更多推荐

pygcn保存模型后加载模型,精度变差

本文发布于:2024-02-14 00:43:49,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1761455.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:模型   精度   加载   pygcn

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!