MCPcopy Create free account
hub / github.com/modAL-python/modAL / test_mc_dropout_bald

Method test_mc_dropout_bald

tests/core_tests.py:752–763  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

750 verbose=1)
751
752 def test_mc_dropout_bald(self):
753 learner = modAL.models.learners.DeepActiveLearner(
754 estimator=self.skorch_classifier,
755 query_strategy=modAL.dropout.mc_dropout_bald,
756 )
757 for random_tie_break in [True, False]:
758 for num_cycles, sample_per_forward_pass in product(range(1, 5), range(1, 5)):
759 for n_samples, n_classes in product(range(1, 5), range(1, 5)):
760 for n_instances in range(1, n_samples):
761 X_pool = torch.randn(n_samples, n_classes)
762 modAL.dropout.mc_dropout_bald(learner, X_pool, n_instances, random_tie_break, [],
763 num_cycles, sample_per_forward_pass)
764
765 def test_mc_dropout_mean_st(self):
766 learner = modAL.models.learners.DeepActiveLearner(

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected