NER项目 P17 测试和预测部分代码修改
上节课,我们在Kaggle上把模型训练好了,并且把模型参数文件下载到了本地,接下来,还需要修改一下测试和预测流程,主要是模型的加载方式有点区别。
代码示例
1、模型加载
内容不可见,请联系管理员开通权限。
2、GPU兼容
内容不可见,请联系管理员开通权限。
3、测试结果
内容不可见,请联系管理员开通权限。
4、模型预测
内容不可见,请联系管理员开通权限。
到目前为止,命名实体识别项目替换数据集的流程,就全部处理完了。我们用了一份好的数据集,训练出的模型效果也比之前的要好得多。在知识图谱项目中,需要用到的disease这一类实体,准确率达到了0.86,已经完全可用了。
最后,还需要把这个模型,封装成一个服务,方便后续在知识图谱项目中调用,这一部分,放到下节课处理。
本文链接:http://ichenhua.cn/edu/note/618
版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!