Spaces:
Running
Running
Update Space (evaluate main: 4f4e17de)
Browse files- spearmanr.py +3 -7
spearmanr.py
CHANGED
|
@@ -113,12 +113,8 @@ class Spearmanr(evaluate.Metric):
|
|
| 113 |
)
|
| 114 |
|
| 115 |
def _compute(self, predictions, references, return_pvalue=False):
|
|
|
|
| 116 |
if return_pvalue:
|
| 117 |
-
return {
|
| 118 |
-
"spearmanr": spearmanr(references, predictions)[0],
|
| 119 |
-
"spearmanr_pvalue": spearmanr(references, predictions)[1],
|
| 120 |
-
}
|
| 121 |
else:
|
| 122 |
-
return {
|
| 123 |
-
"spearmanr": spearmanr(references, predictions)[0],
|
| 124 |
-
}
|
|
|
|
| 113 |
)
|
| 114 |
|
| 115 |
def _compute(self, predictions, references, return_pvalue=False):
|
| 116 |
+
results = spearmanr(references, predictions)
|
| 117 |
if return_pvalue:
|
| 118 |
+
return {"spearmanr": results[0], "spearmanr_pvalue": results[1]}
|
|
|
|
|
|
|
|
|
|
| 119 |
else:
|
| 120 |
+
return {"spearmanr": results[0]}
|
|
|
|
|
|