Add Xgboost regressor

ThomasMeissnerDS · Nov 27, 2023 · 9223337 · 9223337
1 parent 4e4ba73
commit 9223337
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bluecast/ml_modelling/base_classes.py b/bluecast/ml_modelling/base_classes.py
@@ -58,6 +58,6 @@ def predict(self, df: pd.DataFrame) -> np.ndarray:
         """
         Predict on unseen data.
 
-        :return tuple of predicted probabilities and predicted classes
+        :return numpy array of predictions
         """
         pass