p208p2002 / bert-question-answer

基於BERT的中文答題模組

BERT QA

• 基於BERT的問答模組
• 自動滑窗，允許超長文本輸入

Usage

• 安裝相依套件: pip install -r requirements.txt
• 至release下載預訓練模型
• 詳細用法見example.py

# init BertQA
bertQA = BertQA(model = model, tokenizer = tokenizer, device = device)
context="大同國小有三個職員，王大明是校長，張小美是秘書，陳小玉是總務長"

question = "誰是校長"
answer_results = bertQA.ask(context,question)
# score:2.17795 start_index:11(1.07034) end_index:13(1.10761) answer:王大明

question = "陳小玉的工作是什麼"
answer_results = bertQA.ask(context,question)
# score:2.07151 start_index:29(1.84568) end_index:31(0.22583) answer:總務長

Usage

0. 確保已經安裝pytorch1.4+
1. 安裝相依套件: pip install -r requirements.txt
2. 至release下載預訓練模型
3. 將下載的模型放到trained_model/下
4. 詳細用法見example.py

bert-question-answer's Issues

請教您關於fine tune的問題

大大，請問一下，我沒有看到您的程式當中有fine tune的程式碼，只有看到您把資料傳到模型裡面，並不理解這過程當中是如何fine-tune的，可否告知，感謝。