add estimator type

basf · May 4, 2024 · 7ed4b24 · 7ed4b24
1 parent 69134bd
commit 7ed4b24
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/molpipeline/estimators/chemprop/models.py b/molpipeline/estimators/chemprop/models.py
@@ -91,6 +91,19 @@ def classes_(self) -> npt.NDArray[np.int_]:
             raise ValueError("Classes are not set.")
         return self._classes_
 
+    @property
+    def _estimator_type(self) -> str:
+        """Return the estimator type.
+
+        Returns
+        -------
+        str
+            The estimator type.
+        """
+        if self._is_classifier():
+            return "classifier"
+        return "regressor"
+
     def _is_binary_classifier(self) -> bool:
         """Check if the model is a binary classifier.