Skip to content

Commit 2b513c9

Browse files
committed
modAL.models.BayesianOptimizer.teach() signature bug fixed
1 parent 2e22d09 commit 2b513c9

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

modAL/models.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -96,11 +96,11 @@ class BaseLearner(ABC, BaseEstimator):
9696

9797
def __init__(
9898
self,
99-
estimator, # scikit-learner estimator object
99+
estimator, # scikit-learner estimator object
100100
query_strategy=uncertainty_sampling, # callable to query labels
101-
X_training=None, y_training=None, # initial data if available
102-
bootstrap_init=False, # first training with bootstrapping
103-
**fit_kwargs # keyword arguments for fitting the initial data
101+
X_training=None, y_training=None, # initial data if available
102+
bootstrap_init=False, # first training with bootstrapping
103+
**fit_kwargs # keyword arguments for fitting the initial data
104104
):
105105
assert callable(query_strategy), 'query_function must be callable'
106106

@@ -445,7 +445,7 @@ def _set_max(self, y):
445445
if y_max > self.max_val:
446446
self.max_val = y_max
447447

448-
def teach(self):
448+
def teach(self, X, y, bootstrap=False, only_new=False, **fit_kwargs):
449449
"""
450450
Adds X and y to the known training data and retrains the predictor with the
451451
augmented dataset. This method also keeps track of the maximum value encountered

0 commit comments

Comments
 (0)