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)
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)
def gen_parser(parser, model_class): HyperparamConf.gen_parser(parser, model_class, False)
def gen_parser(parser, model_class): parser.add_argument('--multiclass', default=False, action='store_true') HyperparamConf.gen_parser(parser, model_class, False)
def gen_parser(parser, hyperparam_desc): HyperparamConf.gen_parser(parser, hyperparam_desc, False)
def gen_parser(parser, hyperparam_desc): parser.add_argument('--multiclass', default=False, action='store_true') HyperparamConf.gen_parser(parser, hyperparam_desc, True)