Skip to content

Commit cd69a10

Browse files
committed
feat: add estimator type with reg and clf
1 parent 00603e6 commit cd69a10

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

ngboost/api.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,8 @@ def __init__(
9696
validation_fraction,
9797
early_stopping_rounds,
9898
)
99+
100+
self._estimator_type = "regressor"
99101

100102
def __getstate__(self):
101103
state = super().__getstate__()
@@ -172,6 +174,7 @@ def __init__(
172174
tol,
173175
random_state,
174176
)
177+
self._estimator_type = "classifier"
175178

176179
def predict_proba(self, X, max_iter=None):
177180
"""

0 commit comments

Comments
 (0)