Python make_binary_classifier示例

示例#1

文件： test_classifier.py 项目： 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,
     )

示例#2

文件： test_toy.py 项目： 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