IlyaGusev commited on
Commit
6527003
1 Parent(s): a480641

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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/rut5-base-sum-gazeta"
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
- | rut5-base-sum-gazeta | 32.3 | 14.5 | 27.9 | 39.6 | 11.5 |
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/rut5-base-sum-gazeta", list(gazeta_test), "t5_predictions.txt")
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