基于Pytorch的动手学习Bert+FC的文本分类模型
一、模型构建
1.1 模型参数
本次数据集主要是用清华数据集的10分类任务
其中models,放模型文件
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/4045c0bb05bd0d9c071e2bd03b0f3a36.png)
THUCNews下有data文件夹
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/ca8c4e0764530cb974181834435c62a7.png)
data数据类别下有class和test,train、dev数据
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/20e4ecfdbb3c3e469423e95654c6ba17.png)
在models文件夹下,构建bert模型文件
导入相应的库
import torch
import torch.nn as nn
from pytorch_pretrained import BertModel,BertTokenizer