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

代码示例

1、添加验证模块

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

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

2、评估函数修改

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

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

3、边训练边验证

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

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

4、Bert警告 

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

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

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

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