Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
d513beb
1
Parent(s): cd07047
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -530,5 +530,8 @@ class Summarization:
|
|
| 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
|
|
|
|
| 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 |
+
"Rouge_L Mid Precision": results["rougeL"].mid.precision,
|
| 534 |
+
"Rouge_L Mid recall": results["rougeL"].mid.recall,
|
| 535 |
+
"Rouge_L Mid F1": results["rougeL"].mid.fmeasure,
|
| 536 |
}
|
| 537 |
return output
|