基于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的图书分类项目的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。