/taipei-QA-BERT

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

Primary LanguagePython

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

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

檔案說明

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

中文Albert

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

    使用albert-zh-tiny需要對訓練參數進行微調,否則無法獲得好的效果

  • 更多albert-zh模型與用法請參閱p208p2002/albert-zh-for-pytorch-transformers

環境需求

  • python 3.6+
  • pytorch 1.3+