Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
b4a6e7e
1
Parent(s):
d513beb
Update model.py
Browse files- src/models/model.py +12 -0
src/models/model.py
CHANGED
@@ -533,5 +533,17 @@ class Summarization:
|
|
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
|
|
|
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 |
+
"Rouge_L High Precision": results["rougeL"].high.precision,
|
537 |
+
"Rouge_L High recall": results["rougeL"].high.recall,
|
538 |
+
"Rouge_L High F1": results["rougeL"].high.fmeasure,
|
539 |
+
"rougeLsum Low Precision": results["rougeLsum"].low.precision,
|
540 |
+
"rougeLsum Low recall": results["rougeLsum"].low.recall,
|
541 |
+
"rougeLsum Low F1": results["rougeLsum"].low.fmeasure,
|
542 |
+
"rougeLsum Mid Precision": results["rougeLsum"].mid.precision,
|
543 |
+
"rougeLsum Mid recall": results["rougeLsum"].mid.recall,
|
544 |
+
"rougeLsum Mid F1": results["rougeLsum"].mid.fmeasure,
|
545 |
+
"rougeLsum High Precision": results["rougeLsum"].high.precision,
|
546 |
+
"rougeLsum High recall": results["rougeLsum"].high.recall,
|
547 |
+
"rougeLsum High F1": results["rougeLsum"].high.fmeasure,
|
548 |
}
|
549 |
return output
|