fix: template error in s4e6

microsoft · Sep 30, 2024 · d060eb9 · d060eb9
1 parent 1272e32
commit d060eb9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...t/scenarios/kaggle/experiment/playground-series-s4e6_template/model/model_randomforest.py b/...t/scenarios/kaggle/experiment/playground-series-s4e6_template/model/model_randomforest.py
@@ -31,4 +31,4 @@ def predict(model, X):
     Make predictions using the trained model.
     """
     probas = model.predict_proba(X)
-    return np.column_stack([proba[:, 1] for proba in probas])
+    return probas