Update README.md
Browse files
README.md
CHANGED
@@ -52,11 +52,12 @@ print(summary)
|
|
52 |
|
53 |
## Training data
|
54 |
|
55 |
-
- Dataset: https://
|
56 |
|
57 |
## Training procedure
|
58 |
|
59 |
-
- Training script: https://github.com/IlyaGusev/summarus/blob/master/external/hf_scripts/train.py
|
|
|
60 |
|
61 |
## Eval results
|
62 |
|
@@ -132,6 +133,6 @@ gazeta_test = load_dataset('IlyaGusev/gazeta', script_version="v1.0")["test"]
|
|
132 |
predict("IlyaGusev/rut5_base_sum_gazeta", list(gazeta_test), "t5_predictions.txt")
|
133 |
```
|
134 |
|
135 |
-
Evaluation: https://github.com/IlyaGusev/summarus/blob/master/evaluate.py
|
136 |
|
137 |
Flags: --language ru --tokenize-after --lower
|
|
|
52 |
|
53 |
## Training data
|
54 |
|
55 |
+
- Dataset: [Gazeta](https://huggingface.co/datasets/IlyaGusev/gazeta)
|
56 |
|
57 |
## Training procedure
|
58 |
|
59 |
+
- Training script: [train.py](https://github.com/IlyaGusev/summarus/blob/master/external/hf_scripts/train.py)
|
60 |
+
- Config: [t5_training_config.json](https://github.com/IlyaGusev/summarus/blob/master/external/hf_scripts/configs/t5_training_config.json)
|
61 |
|
62 |
## Eval results
|
63 |
|
|
|
133 |
predict("IlyaGusev/rut5_base_sum_gazeta", list(gazeta_test), "t5_predictions.txt")
|
134 |
```
|
135 |
|
136 |
+
Evaluation script: [evaluate.py](https://github.com/IlyaGusev/summarus/blob/master/evaluate.py)
|
137 |
|
138 |
Flags: --language ru --tokenize-after --lower
|