Merge pull request #1319 from perib/estimator_type

estimator type
EpistasisLab · Sep 6, 2023 · 9f02998 · 9f02998
2 parents 881d322 + 56f75d3
commit 9f02998
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tpot/base.py b/tpot/base.py
@@ -2031,6 +2031,9 @@ def classes_(self):
         """The classes labels. Only exist if the last step is a classifier."""
         return self.fitted_pipeline_.classes_
 
+    @property
+    def _estimator_type(self):
+        return self.fitted_pipeline_._estimator_type
 
 def _has_cuml():
     try: