better adding of _estimator_type
@@ -1730,6 +1730,9 @@ class CatBoost(_CatBoostBase):
class CatBoostClassifier(CatBoost):
+
+ _estimator_type = 'classifier'
+
"""
Implementation of the scikit-learn API for CatBoost classification.
@@ -2087,7 +2090,6 @@ class CatBoostClassifier(CatBoost):
if key not in not_params and value is not None:
params[key] = value
- self._estimator_type = 'classifier'
super(CatBoostClassifier, self).__init__(params)
@property
@@ -2355,6 +2357,9 @@ class CatBoostClassifier(CatBoost):
class CatBoostRegressor(CatBoost):
+
+ _estimator_type = 'regressor'
+
"""
Implementation of the scikit-learn API for CatBoost regression.
@@ -2454,7 +2459,6 @@ class CatBoostRegressor(CatBoost):
if key not in not_params and value is not None:
params[key] = value
- self._estimator_type = 'regressor'
super(CatBoostRegressor, self).__init__(params)
def fit(self, X, y=None, cat_features=None, sample_weight=None, baseline=None, use_best_model=None,
