lvwerra HF Staff commited on
Commit
b574800
·
1 Parent(s): e2df10f

Update Space (evaluate main: 4f4e17de)

Browse files
Files changed (1) hide show
  1. 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]}