def parse_args():
    parser = utils.get_arg_parser(description="Train individual pmpm model",
                                  model=True)
    args = utils.parse_args(parser)
    config = Config.from_file(args.config_file)
    config.check_datasets_and_models(models=[args.model])
    return config, args.model
def parse_args():
    parser = utils.get_arg_parser(description='Create metrics',
                                  datasets=True,
                                  model=True,
                                  overwriteable=True)
    args = utils.parse_args(parser)
    config = Config.from_file(args.config_file)
    config.check_datasets_and_models(args.datasets, [args.model])
    return config, args.model, args.datasets, args.overwrite
def parse_args():
    parser = utils.get_arg_parser(description="Create group-level features and"
                                  " target values",
                                  model=True,
                                  datasets=True)
    args = utils.parse_args(parser)
    config = Config.from_file(args.config_file)
    config.check_datasets_and_models(args.datasets)
    return config, args.model, args.datasets
Ejemplo n.º 4
0
def parse_args():
    parser = utils.get_arg_parser(description="Create Matrices",
                                  model=True,
                                  datasets=True,
                                  overwriteable=True)
    parser.add_argument("--full_matrix",
                        action="store_true",
                        help="Create unfiltered full matrices",
                        default=False)
    args = utils.parse_args(parser)
    config = Config.from_file(args.config_file)
    config.check_datasets_and_models(args.datasets)
    return config, args.model, args.datasets, args.full_matrix, args.overwrite
Ejemplo n.º 5
0
def parse_args():
    parser = utils.get_arg_parser(description="Train individual model",
                                  model=True,
                                  overwriteable=True)
    parser.add_argument(
        "--thresholds",
        nargs='+',
        type=float,
        help="list of HiCC thresholds in terms of dollar amount",
        default=[])
    args = utils.parse_args(parser)
    config = Config.from_file(args.config_file)
    config.check_datasets_and_models(models=[args.model])
    return config, args.model, args.thresholds, args.overwrite