项目结构：

文件说明：

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的图书分类项目相关推荐

基于BERT预训练的中文命名实体识别TensorFlow实现
BERT-BiLSMT-CRF-NER Tensorflow solution of NER task Using BiLSTM-CRF model with Google BERT Fine-tun ...
五分钟搭建一个基于BERT的NER模型
BERT 简介 BERT是2018年google 提出来的预训练的语言模型,并且它打破很多NLP领域的任务记录,其提出在nlp的领域具有重要意义.预训练的(pre-train)的语言模型通过无监督的学 ...
基于BERT的化学空间映射
目录背景介绍与方法概述 Related Work 方法概述结果与讨论化学反应分类可视化注意力分布化学空间映射(Mapping the chemical reaction space) 背景介 ...
【文本分类】基于BERT预训练模型的灾害推文分类方法、基于BERT和RNN的新闻文本分类对比
·阅读摘要: 两篇论文,第一篇发表于<图学学报>,<图学学报>是核心期刊:第二篇发表于<北京印刷学院学报>,<北京印刷学院学报>没有任何标签. ·参考文 ...
图书借阅系统java_基于JAVAWEB的图书借阅系统
[实例简介] 基于JAVAWEB的图书借阅系统,包含了源代码.数据库和论文 [实例截图] [核心代码] 图书借阅系统 └── 图书借阅系统 ├── 开题报告.doc ├── 毕设论文.docx ├── ...
java图书馆管理系统_六天写出来的基于Swing的图书管理系统你不来吐槽一下？
前段时间学习完了MySQL和JDBC,想着自己做个东西实战一下,于是写了一个烂大街的图书管理系统...但好歹是自己一个人日夜兼程,硬着头皮做出来的还像样的东西,总结一手吧.看的人欢迎吐槽.我会从以下几 ...
基于bert的语义匹配_构建基于BERT的语义搜索系统…针对“星际迷航”
基于bert的语义匹配 If you read my previous article on Towards Data Science you'll know I'm a bit of a Star ...
bert 多义词_广告行业中那些趣事系列15：超实用的基于BERT美团搜索实践
摘要:本篇主要学习和分析BERT在美团搜索排序中的项目实践.首先介绍了业务背景,主要对比了美团搜索业务场景和我们的应用场景:然后讲了下计算query和doc相关性的方法,主要包括计算字面相关性和计算语 ...

基于Bert的图书分类项目