Update README.md
Browse files
README.md
CHANGED
@@ -1,13 +1,13 @@
|
|
1 |
---
|
2 |
language: ja
|
3 |
widget:
|
4 |
-
- text: "次の出来事の後に起こりうることは何ですか: X
|
5 |
---
|
6 |
|
7 |
# COMET-T5 ja
|
8 |
|
9 |
Finetuned T5 on [ATOMIC ja](https://github.com/nlp-waseda/comet-atomic-ja) using a text-to-text language modeling objective.
|
10 |
-
It was introduced in this paper.
|
11 |
|
12 |
### How to use
|
13 |
|
@@ -18,13 +18,13 @@ Since the generation relies on some randomness, we set a seed for reproducibilit
|
|
18 |
>>> from transformers import pipeline, set_seed
|
19 |
>>> generator = pipeline('text2text-generation', model='nlp-waseda/comet-t5-base-japanese')
|
20 |
>>> set_seed(42)
|
21 |
-
>>> generator("次の出来事の後に起こりうることは何ですか: X
|
22 |
|
23 |
-
[{'generated_text': 'X
|
24 |
-
{'generated_text': 'X
|
25 |
-
{'generated_text': 'X
|
26 |
-
{'generated_text': 'X
|
27 |
-
{'generated_text': 'X
|
28 |
```
|
29 |
|
30 |
### Preprocessing
|
@@ -54,6 +54,6 @@ The model achieves the following results:
|
|
54 |
title = "人間と言語モデルに対するプロンプトを用いたゼロからのイベント常識知識グラフ構築",
|
55 |
booktitle = "言語処理学会第29回年次大会",
|
56 |
year = "2023",
|
57 |
-
url = ""
|
58 |
}
|
59 |
```
|
|
|
1 |
---
|
2 |
language: ja
|
3 |
widget:
|
4 |
+
- text: "次の出来事の後に起こりうることは何ですか: Xがパンを焼く"
|
5 |
---
|
6 |
|
7 |
# COMET-T5 ja
|
8 |
|
9 |
Finetuned T5 on [ATOMIC ja](https://github.com/nlp-waseda/comet-atomic-ja) using a text-to-text language modeling objective.
|
10 |
+
It was introduced in [this paper](https://www.anlp.jp/proceedings/annual_meeting/2023/pdf_dir/B2-5.pdf).
|
11 |
|
12 |
### How to use
|
13 |
|
|
|
18 |
>>> from transformers import pipeline, set_seed
|
19 |
>>> generator = pipeline('text2text-generation', model='nlp-waseda/comet-t5-base-japanese')
|
20 |
>>> set_seed(42)
|
21 |
+
>>> generator("次の出来事の後に起こりうることは何ですか: Xが友人に電話する", max_length=30, num_return_sequences=5, do_sample=True)
|
22 |
|
23 |
+
[{'generated_text': 'Xが友人から返事を得る'},
|
24 |
+
{'generated_text': 'Xが会話する'},
|
25 |
+
{'generated_text': 'Xが友人に怒られる'},
|
26 |
+
{'generated_text': 'Xが退屈しそうな雰囲気になる'},
|
27 |
+
{'generated_text': 'Xが友人と会う'}]
|
28 |
```
|
29 |
|
30 |
### Preprocessing
|
|
|
54 |
title = "人間と言語モデルに対するプロンプトを用いたゼロからのイベント常識知識グラフ構築",
|
55 |
booktitle = "言語処理学会第29回年次大会",
|
56 |
year = "2023",
|
57 |
+
url = "https://www.anlp.jp/proceedings/annual_meeting/2023/pdf_dir/B2-5.pdf"
|
58 |
}
|
59 |
```
|