基于Bert的图书分类项目
项目结构:
文件说明:
data/
数据存放目录
model/
模型存放目录
logs/
日志存放目录
bert/
huggingface的pytorch版的bert的三个文件,分别是:
config.json
pytorch_model.bin
vocab.txt
models/bert.py
模型实现: 在bert后加了个全连接层
config.py
项目配置文件,一些路径配置以及参数配置
dictionary.py
构建word2idx,idx2word字典
train.py
模型训练
predict.py
模型预测
运行
python train.py
python predict.py
数据
链接: https://pan.baidu.com/s/1Uen77gkrCfb36zWVmNPIiQ 密码: 8t0n
下载后将里面的文件放进data目录
源码
总结
如果代码有任何问题,可以私信我。