check 'ensemble' in scores.csv in workflow eval (#682)

2026-08-04 10:47:43 +00:00 · 2025-03-14 17:40:06 +08:00
parent 44d38a5909
commit 641beb2203
2 changed files with 5 additions and 6 deletions
@@ -22,7 +22,7 @@ def print_preds_info(model_name, data_type, preds):
    else:
        print(f"Model {model_name} {data_type} predictions shape: {preds.shape}")

-        if isinstance(preds, pd.DataFrame):
+        if isinstance(preds, (pd.DataFrame, pd.Series)):
            print(preds.head())
        elif isinstance(preds, (np.ndarray, torch.Tensor, tf.Tensor)):
            print(pd.DataFrame(preds).head())