GCN项目 P10 模型测试及多分类评估指标
上一节,完成了模型训练过程,并在训练过程中,缓存了各 Epoch 对应的模型文件,本节就用最后一次训练缓存的模型文件,来对测试数据进行预测。
因为这个项目是一个多分类问题,单纯的看 loss 值,或者整体的准确率都是片面的。所以我们借助 sklearn 的 classification_report() 函数,来查看各个分类的评估指标,从而对模型效果有一个更全面的掌握。
代码示例
1、模型测试
内容不可见,请联系管理员开通权限。
2、多分类评估指标
内容不可见,请联系管理员开通权限。
3、查看分类指标
内容不可见,请联系管理员开通权限。
本文链接:http://ichenhua.cn/edu/note/473
版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!