基于Bert的圖書(shū)分類(lèi)項(xiàng)目

• 項(xiàng)目結(jié)構(gòu)：
• 文件說(shuō)明：
• • data/
  • model/
  • logs/
  • bert/
  • • models/bert.py
  • config.py
  • dictionary.py
  • train.py
  • predict.py
• 運(yùn)行
• 數(shù)據(jù)
• 源碼
• 總結(jié)

項(xiàng)目結(jié)構(gòu)：

文件說(shuō)明：

data/

數(shù)據(jù)存放目錄

model/

模型存放目錄

logs/

日志存放目錄

bert/

huggingface的pytorch版的bert的三個(gè)文件，分別是：
config.json
pytorch_model.bin
vocab.txt

models/bert.py

模型實(shí)現(xiàn): 在bert后加了個(gè)全連接層

config.py

項(xiàng)目配置文件，一些路徑配置以及參數(shù)配置

dictionary.py

構(gòu)建word2idx，idx2word字典

train.py

模型訓(xùn)練

predict.py

模型預(yù)測(cè)

運(yùn)行

python train.py
python predict.py

數(shù)據(jù)

鏈接: https://pan.baidu.com/s/1Uen77gkrCfb36zWVmNPIiQ 密碼: 8t0n

下載后將里面的文件放進(jìn)data目錄

源碼

基于bert的圖書(shū)多分類(lèi)

總結(jié)

如果代碼有任何問(wèn)題，可以私信我。

總結(jié)

以上是生活随笔為你收集整理的基于Bert的图书分类项目的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。