Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,9 @@ Training Procedure: 6 epochs, all details and hyperparameters in [this Google Co
|
|
25 |
|
26 |
## Testing Metrics
|
27 |
|
28 |
-
Rouge1: 40.24
|
|
|
|
|
29 |
|
30 |
## How to Use
|
31 |
|
@@ -36,7 +38,8 @@ model_name = "wanderer-msk/ruT5-base_headline_generation"
|
|
36 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
37 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
38 |
|
39 |
-
news_text = "Земляне продолжают осваивать Марс.
|
|
|
40 |
|
41 |
model_input = tokenizer(
|
42 |
news_text,
|
|
|
25 |
|
26 |
## Testing Metrics
|
27 |
|
28 |
+
- Rouge1: 40.24
|
29 |
+
- Rouge2: 23.05
|
30 |
+
- RougeL: 37.57
|
31 |
|
32 |
## How to Use
|
33 |
|
|
|
38 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
39 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
40 |
|
41 |
+
news_text = """Земляне продолжают осваивать Марс.
|
42 |
+
Колонисты уже посадили на красной планете 42 яблони."""
|
43 |
|
44 |
model_input = tokenizer(
|
45 |
news_text,
|