Maria Castellanos commited on
Commit
ec86576
·
1 Parent(s): 18865de

Add check for eval std

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -120,8 +120,8 @@ def metrics_per_ep(pred: np.ndarray,
120
  spr = np.nan
121
  ktau = np.nan
122
  else:
123
- spr, _ = spearmanr(true, pred)
124
- ktau, _ = kendalltau(true, pred)
125
 
126
  return mae, rae, r2, spr, ktau
127
 
 
120
  spr = np.nan
121
  ktau = np.nan
122
  else:
123
+ spr = spearmanr(true, pred).statistic
124
+ ktau = kendalltau(true, pred).statistic
125
 
126
  return mae, rae, r2, spr, ktau
127