Add Xgboost regressor

ThomasMeissnerDS · Nov 27, 2023 · 4e4ba73 · 4e4ba73
1 parent 76c8365
commit 4e4ba73
Show file tree

Hide file tree

Showing 2 changed files with 831 additions and 0 deletions.
diff --git a/bluecast/ml_modelling/base_classes.py b/bluecast/ml_modelling/base_classes.py
@@ -34,3 +34,30 @@ def predict(self, df: pd.DataFrame) -> Tuple[PredictedProbas, PredictedClasses]:
         :return tuple of predicted probabilities and predicted classes
         """
         pass
+
+
+class BaseClassMlRegressionModel(ABC):
+    """Base class for all ML models.
+
+    Enforces the implementation of the fit and predict methods.
+    If hyperparameter tuning is required, then the fit method should implement the tuning.
+    """
+
+    @abstractmethod
+    def fit(
+        self,
+        x_train: pd.DataFrame,
+        x_test: pd.DataFrame,
+        y_train: pd.Series,
+        y_test: pd.Series,
+    ) -> Optional[Any]:
+        pass
+
+    @abstractmethod
+    def predict(self, df: pd.DataFrame) -> np.ndarray:
+        """
+        Predict on unseen data.
+
+        :return tuple of predicted probabilities and predicted classes
+        """
+        pass