Maria Castellanos
commited on
Commit
·
ec86576
1
Parent(s):
18865de
Add check for eval std
Browse files
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
|
| 124 |
-
ktau
|
| 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 |
|