예제 #1
0
 def gen_parser(parser, model_class, multiclass=True):
     if multiclass:
         parser.add_argument('--multiclass', default=False,
                             action='store_true',
                             help='The classifier is trained on the '
                                  'families instead of the binary labels.')
     HyperparamConf.gen_parser(parser, model_class, False)
예제 #2
0
 def gen_main_model_parser(parser):
     group = parser.add_argument_group('Classification model parameters')
     models = classifiers.get_factory().get_methods(supervised=True)
     group.add_argument('--model-class',
                        choices=models,
                        default='LogisticRegression',
                        help='Model class trained at each iteration. '
                        'Default: LogisticRegression.')
     HyperparamConf.gen_parser(group, None, True, subgroup=False)
예제 #3
0
 def gen_parser(parser, model_class):
     HyperparamConf.gen_parser(parser, model_class, False)
예제 #4
0
파일: __init__.py 프로젝트: tuantmb/SecuML
 def gen_parser(parser, model_class):
     parser.add_argument('--multiclass', default=False, action='store_true')
     HyperparamConf.gen_parser(parser, model_class, False)
예제 #5
0
 def gen_parser(parser, hyperparam_desc):
     HyperparamConf.gen_parser(parser, hyperparam_desc, False)
예제 #6
0
 def gen_parser(parser, hyperparam_desc):
     parser.add_argument('--multiclass', default=False, action='store_true')
     HyperparamConf.gen_parser(parser, hyperparam_desc, True)