Aug 10

chenyangkang · Aug 10, 2023 · d6647da · d6647da
1 parent 6676a85
commit d6647da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/BirdSTEM/model/AdaSTEM.py b/BirdSTEM/model/AdaSTEM.py
@@ -507,8 +507,8 @@ def eval_STEM_res(self, task, y_test, y_pred, cls_threashold=None):
                 'y_ture':y_test,
                 'pred':y_pred
             }).dropna()
-            s_r, _ = spearmanr(a.y_ture, a.pred)
-            p_r, _ = pearsonr(a.y_ture, a.pred)
+            s_r, _ = spearmanr(np.array(a.y_ture), np.array(a.pred))
+            p_r, _ = pearsonr(np.array(a.y_ture), np.array(a.pred))
             r2 = r2_score(a.y_ture, a.pred)
             MAE = mean_absolute_error(a.y_ture, a.pred)
             MSE = mean_squared_error(a.y_ture, a.pred)