Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
2981c19
1
Parent(s):
ab326e4
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
@@ -510,5 +510,8 @@ class Summarization:
|
|
510 |
|
511 |
output = {
|
512 |
"Rouge_1 Low Precision": results["rouge1"].low.precision,
|
|
|
|
|
|
|
513 |
}
|
514 |
return output
|
|
|
510 |
|
511 |
output = {
|
512 |
"Rouge_1 Low Precision": results["rouge1"].low.precision,
|
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
|