Python make_binary_classifier Examples

Example #1

File: test_classifier.py Project: yashgupta-7/skorch

 def module_cls(self):
     from skorch.toy import make_binary_classifier
     return make_binary_classifier(
         input_units=20,
         hidden_units=10,
         output_units=1,
         num_hidden=1,
         dropout=0,
     )

Example #2

File: test_toy.py Project: MakotoDataiku/lstm-test

 def test_make_binary_classifier(self):
     from skorch.toy import make_binary_classifier
     module = make_binary_classifier()()
     assert isinstance(module.sequential[-1], nn.Linear)
     assert module.squeeze_output is True