argparse_example.py

import argparse

from models.model import model_class, model_type
from libs.utils import run_train


"""
Main
"""


def main():
    parser = argparse.ArgumentParser(description='Process some integers.')
    parser.add_argument('--dataset', metavar='data_path',
                        type=str, help='path of train csv file')
    parser.add_argument('--param', metavar='param_path',
                        type=str, help='path of param json file')
    parser.add_argument('--model', metavar='model_path',
                        type=str, help='path of model')
    parser.add_argument('--result', metavar='result_path',
                        type=str, help='path of result')
    p = parser.parse_args()

    print("{}\ntrain start\n{}".format('*'*30, '*'*30))
    run_train(model_class=model_class,
              model_type=model_type,
              data_path=p.dataset,
              model_path=p.model,
              param_path=p.param,
              result_path=p.result)


if __name__ == '__main__':
    main()