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
|