Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
6c8b15e
1
Parent(s):
c1f35ff
Update model.py
Browse files- src/models/model.py +6 -0
src/models/model.py
CHANGED
@@ -474,3 +474,9 @@ class Summarization:
|
|
474 |
'Rouge_1 Mid F1': results["rouge1"].mid.fmeasure,
|
475 |
'Rouge_1 High Precision': results["rouge1"].high.precision,
|
476 |
'Rouge_1 High recall': results["rouge1"].high.recall,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
474 |
'Rouge_1 Mid F1': results["rouge1"].mid.fmeasure,
|
475 |
'Rouge_1 High Precision': results["rouge1"].high.precision,
|
476 |
'Rouge_1 High recall': results["rouge1"].high.recall,
|
477 |
+
'Rouge_1 High F1': results["rouge1"].high.fmeasure,
|
478 |
+
},
|
479 |
+
'Rouge 2': {
|
480 |
+
'Rouge_2 Low Precision': results["rouge2"].low.precision,
|
481 |
+
'Rouge_2 Low recall': results["rouge2"].low.recall,
|
482 |
+
'Rouge_2 Low F1': results["rouge2"].low.fmeasure,
|