Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
d7bb16b
1
Parent(s): 6c6d23b
Update model.py
Browse files- src/models/model.py +1 -1
src/models/model.py
CHANGED
|
@@ -509,7 +509,6 @@ class Summarization:
|
|
| 509 |
results = metric.compute(predictions=predictions, references=references)
|
| 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,
|
|
@@ -552,5 +551,6 @@ class Summarization:
|
|
| 552 |
"rougeLsum High recall": results["rougeLsum"].high.recall,
|
| 553 |
"rougeLsum High F1": results["rougeLsum"].high.fmeasure,
|
| 554 |
},
|
|
|
|
| 555 |
}
|
| 556 |
return output
|
|
|
|
| 509 |
results = metric.compute(predictions=predictions, references=references)
|
| 510 |
|
| 511 |
output = {
|
|
|
|
| 512 |
"Rouge_1 Low recall": results["rouge1"].low.recall,
|
| 513 |
"Rouge_1 Low F1": results["rouge1"].low.fmeasure,
|
| 514 |
"Rouge_1 Mid Precision": results["rouge1"].mid.precision,
|
|
|
|
| 551 |
"rougeLsum High recall": results["rougeLsum"].high.recall,
|
| 552 |
"rougeLsum High F1": results["rougeLsum"].high.fmeasure,
|
| 553 |
},
|
| 554 |
+
"Rouge_1 Low Precision": results["rouge1"].low.precision,
|
| 555 |
}
|
| 556 |
return output
|