台北QA問答機器人(使用BERT、ALBERT)
問一個問題,告訴你應該去哪個單位處理這些問題
train.py
predict.py
使用albert-zh-tiny需要對訓練參數進行微調,否則無法獲得好的效果