Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
1875c38
1
Parent(s):
ac42dfa
Update model.py
Browse files- src/models/model.py +4 -0
src/models/model.py
CHANGED
@@ -522,5 +522,9 @@ class Summarization:
|
|
522 |
"Rouge_2 Low recall": results["rouge2"].low.recall,
|
523 |
"Rouge_2 Low F1": results["rouge2"].low.fmeasure,
|
524 |
"Rouge_2 Mid Precision": results["rouge2"].mid.precision,
|
|
|
|
|
|
|
|
|
525 |
}
|
526 |
return output
|
|
|
522 |
"Rouge_2 Low recall": results["rouge2"].low.recall,
|
523 |
"Rouge_2 Low F1": results["rouge2"].low.fmeasure,
|
524 |
"Rouge_2 Mid Precision": results["rouge2"].mid.precision,
|
525 |
+
"Rouge_2 Mid recall": results["rouge2"].mid.recall,
|
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
|