基于Pytorch的动手学习Bert+FC的文本分类模型 一、模型构建 1.1 模型参数 1.2 构建Bert+FC模型 二、辅助模块 2.1 数据构建函数 2.2 数据Batch迭代器 三、模型训练 3.1 模型训练函数 3.2 模型结果评估函数 四、主函数 4.1 主函数 4.2 完整代码 一、模型构建 1.1 模型参数 本次数据集主要是用清华数据集的10分类任务 其中models,放模型文件 THUCNews下有data文件夹 data数据类别下有class和test,train、dev数据 在models文件夹下,构建bert模型文件 导入相应的库 import torch import torch.nn