Skip to content

台北QA問答機器人(使用BERT、ALBERT)

Notifications You must be signed in to change notification settings

deternan/taipei-QA-BERT

 
 

Repository files navigation

台北QA問答機器人(with BERT)

問一個問題,告訴你應該去哪個單位處理這些問題

檔案說明

  • train.py : 模型訓練(BERT fine-tune)
  • predict.py : 提問預測
  • tutorial : 投影片檔案

中文Albert

  • 預設使用bert-base-chinese
  • 欲切換至albert-zh-tiny請將train.pypredict.py對應註解拿掉

環境需求

  • python 3.6+
  • pytorch 1.3+

About

台北QA問答機器人(使用BERT、ALBERT)

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 83.2%
  • Jupyter Notebook 16.8%