Update README.md
Browse files
README.md
CHANGED
@@ -26,7 +26,7 @@ Colab: [link](https://colab.research.google.com/drive/1re5E26ZIDUpAx1gOCZkbF3hcw
|
|
26 |
```python
|
27 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
28 |
|
29 |
-
model_name = "IlyaGusev/
|
30 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
31 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
32 |
|
@@ -62,7 +62,7 @@ print(summary)
|
|
62 |
|
63 |
| Model | R-1-f | R-2-f | R-L-f | chrF | BLEU |
|
64 |
|:--------------------------|:------|:------|:------|:-----|:-----|
|
65 |
-
|
|
66 |
|
67 |
Predicting all summaries:
|
68 |
```python
|
@@ -119,7 +119,7 @@ def predict(
|
|
119 |
w.write(p.strip().replace("\n", " ") + "\n")
|
120 |
|
121 |
gazeta_test = load_dataset('IlyaGusev/gazeta', script_version="v1.0")["test"]
|
122 |
-
predict("IlyaGusev/
|
123 |
```
|
124 |
|
125 |
Evaluation: https://github.com/IlyaGusev/summarus/blob/master/evaluate.py
|
|
|
26 |
```python
|
27 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
28 |
|
29 |
+
model_name = "IlyaGusev/rut5_base_sum_gazeta"
|
30 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
31 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
32 |
|
|
|
62 |
|
63 |
| Model | R-1-f | R-2-f | R-L-f | chrF | BLEU |
|
64 |
|:--------------------------|:------|:------|:------|:-----|:-----|
|
65 |
+
| rut5_base_sum_gazeta | 32.3 | 14.5 | 27.9 | 39.6 | 11.5 |
|
66 |
|
67 |
Predicting all summaries:
|
68 |
```python
|
|
|
119 |
w.write(p.strip().replace("\n", " ") + "\n")
|
120 |
|
121 |
gazeta_test = load_dataset('IlyaGusev/gazeta', script_version="v1.0")["test"]
|
122 |
+
predict("IlyaGusev/rut5_base_sum_gazeta", list(gazeta_test), "t5_predictions.txt")
|
123 |
```
|
124 |
|
125 |
Evaluation: https://github.com/IlyaGusev/summarus/blob/master/evaluate.py
|