基于pytorch的中文文本分类
采用Pytorch搭建Bert模型
训练数据使用THUCNews新闻文本分类数据集的子集。
BIT《自然语言理解初步》课程大作业2
- chinese_roberta_wwm_ext_pytorch/为预训练的Bert模型
- data/ckpt存储训练好的模型
- data/cnews文件夹存储经过处理的THUCNews新闻文本分类数据集
- BIT2022_NLU_h2.ipynb文件用于功能演示。
基于pytorch的中文文本分类
采用Pytorch搭建Bert模型
训练数据使用THUCNews新闻文本分类数据集的子集。
BIT《自然语言理解初步》课程大作业2