@@ -642,7 +642,7 @@ def test_keras(self):
642
642
643
643
def test_sklearn (self ):
644
644
learner = modAL .models .learners .ActiveLearner (
645
- estimator = RandomForestClassifier (),
645
+ estimator = RandomForestClassifier (n_estimators = 10 ),
646
646
X_training = np .random .rand (10 , 10 ),
647
647
y_training = np .random .randint (0 , 2 , size = (10 ,))
648
648
)
@@ -667,7 +667,7 @@ def test_sparse_matrices(self):
667
667
initial_idx = np .random .choice (range (n_samples ), size = 5 , replace = False )
668
668
669
669
learner = modAL .models .learners .ActiveLearner (
670
- estimator = RandomForestClassifier (), query_strategy = query_strategy ,
670
+ estimator = RandomForestClassifier (n_estimators = 10 ), query_strategy = query_strategy ,
671
671
X_training = X_pool [initial_idx ], y_training = y_pool [initial_idx ]
672
672
)
673
673
query_idx , query_inst = learner .query (X_pool )
@@ -949,7 +949,7 @@ def test_strategies(self):
949
949
X_training = np .random .rand (n_pool_instances , 5 )
950
950
y_training = np .random .randint (0 , 2 , size = (n_pool_instances , n_classes ))
951
951
X_pool = np .random .rand (n_pool_instances , 5 )
952
- classifier = OneVsRestClassifier (SVC (probability = True ))
952
+ classifier = OneVsRestClassifier (SVC (probability = True , gamma = 'auto' ))
953
953
classifier .fit (X_training , y_training )
954
954
modAL .multilabel .mean_max_loss (classifier , X_pool , n_query_instances )
955
955
modAL .multilabel .max_loss (classifier , X_pool , n_query_instances )
0 commit comments