GCN项目 P9 加载训练数据并训练模型
目前,我们已经搭建好了完整的图卷积模型,接下来,就可以加载数据,并进行模型训练了。
虽然模型是基于图结构,但最终目的,还是解决节点分类的问题,所以损失函数就用 CrossEntropyLoss 即可。
另外,还有一个需要注意的问题,就是在建立图结构过程中,可能会产生干扰的孤立节点。为避免孤立点导致的矩阵形状不能对齐的问题,需要在进模型之前,将孤立节点移除。
代码示例
1、创建训练文件
内容不可见,请联系管理员开通权限。
2、加载数据
内容不可见,请联系管理员开通权限。
3、模型训练
内容不可见,请联系管理员开通权限。
本文链接:http://www.ichenhua.cn/edu/note/472
版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!