/BERT-TextClassification

使用BERT预训练模型实现中文短文本的分类

Primary LanguagePythonMIT LicenseMIT

BERT-TextClassification

使用BERT预训练模型实现中文短文本的分类


请自行下载模型文件

bert_pretrain/ pytorch_model.bin
bert_config.json
vocab.txt

下载地址: https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-chinese.tar.gz
若名称不符请自行修改。


THUCNews/saved_dict/ bert.ckpt

下载地址: 链接:https://pan.baidu.com/s/1TDXtwtHQf8CYwmzI5_lNVA?pwd=x826 提取码:x826

自行预训练

请参考:https://github.com/649453932/Bert-Chinese-Text-Classification-Pytorch