Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
cd07047
1
Parent(s):
1875c38
Update model.py
Browse files- src/models/model.py +4 -0
src/models/model.py
CHANGED
|
@@ -526,5 +526,9 @@ class Summarization:
|
|
| 526 |
"Rouge_2 Mid F1": results["rouge2"].mid.fmeasure,
|
| 527 |
"Rouge_2 High Precision": results["rouge2"].high.precision,
|
| 528 |
"Rouge_2 High recall": results["rouge2"].high.recall,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 529 |
}
|
| 530 |
return output
|
|
|
|
| 526 |
"Rouge_2 Mid F1": results["rouge2"].mid.fmeasure,
|
| 527 |
"Rouge_2 High Precision": results["rouge2"].high.precision,
|
| 528 |
"Rouge_2 High recall": results["rouge2"].high.recall,
|
| 529 |
+
"Rouge_2 High F1": results["rouge2"].high.fmeasure,
|
| 530 |
+
"Rouge_L Low Precision": results["rougeL"].low.precision,
|
| 531 |
+
"Rouge_L Low recall": results["rougeL"].low.recall,
|
| 532 |
+
"Rouge_L Low F1": results["rougeL"].low.fmeasure,
|
| 533 |
}
|
| 534 |
return output
|