Update README.md
Browse files
README.md
CHANGED
@@ -26,5 +26,19 @@ metrics:
|
|
26 |
|
27 |
將問題(Question)和10句各自獨立的句子(Sentences)組合成輸入,模型可以推理出這10個句子中和問題最相符的答案。
|
28 |
|
29 |
-
|
30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
將問題(Question)和10句各自獨立的句子(Sentences)組合成輸入,模型可以推理出這10個句子中和問題最相符的答案。
|
28 |
|
29 |
+
```python
|
30 |
+
\\Input=question:balabal context:senten1[SEP]senten2[SEP]senten3....
|
31 |
+
\\Output=abstract result
|
32 |
+
model = MT5ForConditionalGeneration.from_pretrained("nchu-nlp-lct/nlp-mt5-base-drcd")
|
33 |
+
tokenizer = MT5Tokenizer.from_pretrained("nchu-nlp-lct/nlp-mt5-base-drcd")
|
34 |
+
|
35 |
+
tokenized_inputs = tokenizer.batch_encode_plus(
|
36 |
+
["confusion:2022年,你知道日本職棒上季僅有24個打席,卻奪下年度盜壘王獎項的球員是誰嗎? context:和田康士朗,2021年-- 全季出賽96場,且僅有24打席、19打數,但季末仍以24次盜壘成功與荻野貴司、西川遥輝、源田壮亮等3人並列洋聯盜壘王,為2011年藤村大介後再次出現未達規定打席的盜壘王;且一舉刷新1966年山本公士158打席、1944年吳昌征93打席的兩聯盟暨單一聯盟最少打席獲得盜壘王的雙重紀錄,並為首位獲得日本職棒年度個人獎項的棒球挑戰聯盟出身球員[SEP]和田康士朗,個人年表,2021年-- 全季出賽96場,且僅有24打席、19打數,但季末仍以24次盜壘成功與荻野貴司、西川遥輝、源田壮亮等3人並列洋聯盜壘王,為2011年藤村大介後再次出現未達規定打席的盜壘王;且一舉刷新1966年山本公士158打席、1944年吳昌征93打席的兩聯盟暨單一聯盟最少打席獲得盜壘王的雙重紀錄,並為首位獲得日本職棒年度個人獎項的棒球挑戰聯盟出身球員[SEP]日本職棒年度盜壘王,單一聯盟時期,年度:1944,球員名:吳新亨,球隊:巨人隊,盜壘成功:19,盜壘次數:24,成功率:.792"], pad_to_max_length=True, return_tensors="pt"
|
37 |
+
)
|
38 |
+
out=model.generate(tokenized_inputs['input_ids'])
|
39 |
+
|
40 |
+
//取得答案
|
41 |
+
for ids in out:
|
42 |
+
print(tokenizer2.decode(ids))
|
43 |
+
|
44 |
+
```
|