args.py

from argparse import ArgumentParser


def read_args():
    parser = ArgumentParser()

    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument(
        "-d", "--data", dest="data_path", help="path to preprocessed dataset"
    )
    group.add_argument(
        "-l",
        "--load_path",
        dest="load_path",
        help="path to load model files",
        metavar="FILE",
    )

    parser.add_argument(
        "-m",
        "--model_path",
        dest="model_path",
        help="path to save and load checkpoints",
        metavar="FILE",
        required=False,
    )
    parser.add_argument(
        "-s",
        "--save_path",
        dest="save_path",
        help="path to save model files",
        metavar="FILE",
        required=False,
    )

    parser.add_argument(
        "-t",
        "--test",
        dest="test_path",
        help="path to test file",
        metavar="FILE",
        required=False,
    )

    parser.add_argument(
        "-p",
        "--predict",
        dest="predict",
        type=str,
        default="",
        help='starts prediction mode, argument is "cpp" or "java" dependin on language model',
    )
    parser.add_argument(
        "-c",
        "--continue_training_from_checkpoint",
        dest="continue_from_checkpoint",
        type=str,
        help='Continue training model from a previous checkpoint, if it exists.',
    )

    parser.add_argument("--debug", action="store_true")
    parser.add_argument("--seed", type=int, default=239)
    return parser.parse_args()