Skip to content

Commit cad9cc4

Browse files
committed
fix: imports from modAL.models.learners fixed
1 parent 6785bd3 commit cad9cc4

File tree

2 files changed

+18
-1
lines changed

2 files changed

+18
-1
lines changed

modAL/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from .models import ActiveLearner, Committee, CommitteeRegressor
1+
from .models.learners import ActiveLearner, Committee, CommitteeRegressor
22
from .acquisition import PI, EI, UCB, optimizer_PI, optimizer_EI, optimizer_UCB, max_PI, max_EI, max_UCB
33
from .uncertainty import classifier_uncertainty, classifier_margin, classifier_entropy, \
44
uncertainty_sampling, margin_sampling, entropy_sampling

modAL/models/learners.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,11 @@
1212
from modAL.disagreement import vote_entropy_sampling, max_std_sampling
1313
from modAL.acquisition import max_EI
1414

15+
"""
16+
Classes for active learning algorithms
17+
--------------------------------------
18+
"""
19+
1520

1621
class ActiveLearner(BaseLearner):
1722
"""
@@ -94,6 +99,12 @@ def teach(self, X: modALinput, y: modALinput, bootstrap: bool = False, only_new:
9499
self._fit_on_new(X, y, bootstrap=bootstrap, **fit_kwargs)
95100

96101

102+
"""
103+
Classes for Bayesian optimization
104+
---------------------------------
105+
"""
106+
107+
97108
class BayesianOptimizer(BaseLearner):
98109
"""
99110
This class is an abstract model of a Bayesian optimizer algorithm.
@@ -219,6 +230,12 @@ def teach(self, X: modALinput, y: modALinput, bootstrap: bool = False, only_new:
219230
self._set_max(X, y)
220231

221232

233+
"""
234+
Classes for committee based algorithms
235+
--------------------------------------
236+
"""
237+
238+
222239
class Committee(BaseCommittee):
223240
"""
224241
This class is an abstract model of a committee-based active learning algorithm.

0 commit comments

Comments
 (0)