基于Bert的图书分类项目

该项目采用Bert模型进行图书分类,包括数据预处理、模型训练和预测。数据集可在提供的百度网盘链接中获取,放入data目录。模型在Bert基础上添加了全连接层,配置文件在config.py,字典构建在dictionary.py,训练通过train.py完成,预测使用predict.py。项目结构清晰,便于理解和复用。
摘要由CSDN通过智能技术生成

项目结构:

在这里插入图片描述

文件说明:

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的图书多分类

总结

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

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ToTensor

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值