Martin Dočekal
commited on
Commit
·
d2429ca
1
Parent(s):
d8248d0
example fix
Browse files- README.md +3 -3
- rouge_raw.py +3 -3
README.md
CHANGED
|
@@ -34,7 +34,7 @@ predictions = ["the cat is on the mat", "hello there"]
|
|
| 34 |
references = ["the cat is on the mat", "hello there"]
|
| 35 |
results = rougeraw.compute(predictions=predictions, references=references)
|
| 36 |
print(results)
|
| 37 |
-
{'
|
| 38 |
```
|
| 39 |
|
| 40 |
|
|
@@ -50,13 +50,13 @@ There are precision, recall, F1 values for rougeraw-1, rougeraw-2 and rougeraw-l
|
|
| 50 |
|
| 51 |
Key format:
|
| 52 |
```
|
| 53 |
-
{1|2|
|
| 54 |
e.g.: 1_low_precision
|
| 55 |
```
|
| 56 |
|
| 57 |
If aggregate is False the format is:
|
| 58 |
```
|
| 59 |
-
{1|2|
|
| 60 |
e.g.: 1_precision
|
| 61 |
```
|
| 62 |
|
|
|
|
| 34 |
references = ["the cat is on the mat", "hello there"]
|
| 35 |
results = rougeraw.compute(predictions=predictions, references=references)
|
| 36 |
print(results)
|
| 37 |
+
{'1_low_precision': 1.0, '1_low_recall': 1.0, '1_low_fmeasure': 1.0, '1_mid_precision': 1.0, '1_mid_recall': 1.0, '1_mid_fmeasure': 1.0, '1_high_precision': 1.0, '1_high_recall': 1.0, '1_high_fmeasure': 1.0, '2_low_precision': 1.0, '2_low_recall': 1.0, '2_low_fmeasure': 1.0, '2_mid_precision': 1.0, '2_mid_recall': 1.0, '2_mid_fmeasure': 1.0, '2_high_precision': 1.0, '2_high_recall': 1.0, '2_high_fmeasure': 1.0, 'L_low_precision': 1.0, 'L_low_recall': 1.0, 'L_low_fmeasure': 1.0, 'L_mid_precision': 1.0, 'L_mid_recall': 1.0, 'L_mid_fmeasure': 1.0, 'L_high_precision': 1.0, 'L_high_recall': 1.0, 'L_high_fmeasure': 1.0}
|
| 38 |
```
|
| 39 |
|
| 40 |
|
|
|
|
| 50 |
|
| 51 |
Key format:
|
| 52 |
```
|
| 53 |
+
{1|2|L}_{low|mid|high}_{precision|recall|fmeasure}
|
| 54 |
e.g.: 1_low_precision
|
| 55 |
```
|
| 56 |
|
| 57 |
If aggregate is False the format is:
|
| 58 |
```
|
| 59 |
+
{1|2|L}_{precision|recall|fmeasure}
|
| 60 |
e.g.: 1_precision
|
| 61 |
```
|
| 62 |
|
rouge_raw.py
CHANGED
|
@@ -380,9 +380,9 @@ class RougeRaw(evaluate.Metric):
|
|
| 380 |
"2_precision": res["2"].p,
|
| 381 |
"2_recall": res["2"].r,
|
| 382 |
"2_fmeasure": res["2"].f,
|
| 383 |
-
"
|
| 384 |
-
"
|
| 385 |
-
"
|
| 386 |
}
|
| 387 |
|
| 388 |
if select is not None:
|
|
|
|
| 380 |
"2_precision": res["2"].p,
|
| 381 |
"2_recall": res["2"].r,
|
| 382 |
"2_fmeasure": res["2"].f,
|
| 383 |
+
"L_precision": res["L"].p,
|
| 384 |
+
"L_recall": res["L"].r,
|
| 385 |
+
"L_fmeasure": res["L"].f,
|
| 386 |
}
|
| 387 |
|
| 388 |
if select is not None:
|