基于Bert的图书分类项目

• 项目结构：
• 文件说明：
• • data/
  • model/
  • logs/
  • bert/
  • • models/bert.py
  • config.py
  • dictionary.py
  • train.py
  • predict.py
• 运行
• 数据
• 源码
• 总结

项目结构：

文件说明：

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目录

源码

基于bert的图书多分类

总结

如果代码有任何问题，可以私信我。

总结

以上是生活随笔为你收集整理的基于Bert的图书分类项目的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。