/taipei-QA-BERT

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

Primary LanguagePython

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

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

檔案說明

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

環境需求

  • python 3.6+
  • pytorch 1.3+