之前给大家讲的实体识别项目里,在训练过程中,只输出了loss值。其实只有loss值是看不出模型好坏的,正好这份数据有验证集,我们把验证的流程,加到训练过程中,就可以直观的看到模型效果的变化。

代码示例

1、添加验证模块

验证和测试流程类似,可以复制test.py文件进行修改。BATCH_SIZE 添加统一配置。

内容不可见,请联系管理员开通权限。

2、评估函数修改

测试环节看整体效果,验证环节看f1-score。

内容不可见,请联系管理员开通权限。

3、边训练边验证

f1-score达到0.6,保存模型参数。

内容不可见,请联系管理员开通权限。

4、Bert警告

内容不可见,请联系管理员开通权限。

现在训练部分的代码已经改完了,接下来还是老办法,把模型放到Kaggle上,用免费的GPU资源,来训练模型。

本文链接:http://www.ichenhua.cn/edu/note/616

版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!