Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
cf02d2a
1
Parent(s):
1cf9cc4
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
@@ -516,5 +516,8 @@ class Summarization:
|
|
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
|
|
|
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 |
+
"Rouge_1 High recall": results["rouge1"].high.recall,
|
520 |
+
"Rouge_1 High F1": results["rouge1"].high.fmeasure,
|
521 |
+
"Rouge_2 Low Precision": results["rouge2"].low.precision,
|
522 |
}
|
523 |
return output
|