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

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

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

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

改成列表就好了。