UNIST-Eunchan
commited on
Commit
•
1a89701
1
Parent(s):
6558c20
Update README.md
Browse files
README.md
CHANGED
@@ -94,6 +94,13 @@ Paper Summarization
|
|
94 |
attention_mask=inputs["attention_mask"].to(device),
|
95 |
num_beam_groups=5,diversity_penalty=1.0,num_beams=5,min_length=150,max_length=128*4)
|
96 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
97 |
|
98 |
|
99 |
|
|
|
94 |
attention_mask=inputs["attention_mask"].to(device),
|
95 |
num_beam_groups=5,diversity_penalty=1.0,num_beams=5,min_length=150,max_length=128*4)
|
96 |
```
|
97 |
+
- R-1 | R-2 | R-L | R-LSUM : 43.3017 | 16.6023 | 24.1867 | 33.7019 at
|
98 |
+
```(python)
|
99 |
+
model.generate(input_ids =inputs["input_ids"].to(device),
|
100 |
+
attention_mask=inputs["attention_mask"].to(device),
|
101 |
+
length_penalty=1.2, num_beams=4, max_length=128*4,min_length=150, no_repeat_ngram_size= 3, temperature=0.9,top_k=50,top_p=0.92)
|
102 |
+
|
103 |
+
```
|
104 |
|
105 |
|
106 |
|