Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
1cf9cc4
1
Parent(s): 2981c19
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -513,5 +513,8 @@ class Summarization:
|
|
| 513 |
"Rouge_1 Low recall": results["rouge1"].low.recall,
|
| 514 |
"Rouge_1 Low F1": results["rouge1"].low.fmeasure,
|
| 515 |
"Rouge_1 Mid Precision": results["rouge1"].mid.precision,
|
|
|
|
|
|
|
|
|
|
| 516 |
}
|
| 517 |
return output
|
|
|
|
| 513 |
"Rouge_1 Low recall": results["rouge1"].low.recall,
|
| 514 |
"Rouge_1 Low F1": results["rouge1"].low.fmeasure,
|
| 515 |
"Rouge_1 Mid Precision": results["rouge1"].mid.precision,
|
| 516 |
+
"Rouge_1 Mid recall": results["rouge1"].mid.recall,
|
| 517 |
+
"Rouge_1 Mid F1": results["rouge1"].mid.fmeasure,
|
| 518 |
+
"Rouge_1 High Precision": results["rouge1"].high.precision,
|
| 519 |
}
|
| 520 |
return output
|