Train with argparse

train_argparse.py

import argparse


def train(data_path, model_path, learning_rate, batch_size):
    ...
 

if __name__ == "__main__":
    argparser = argparse.ArgumentParser()
    argparser.add_argument("--data_path", type=str, help="path to train data")
    argparser.add_argument("--model_path", type=str, help="path to save the model")
    argparser.add_argument("--learning_rate", type=float, help="learning rate param")
    argparser.add_argument("--batch_size", type=int, help="batch size param")
    args = argparser.parse_args()

    data_path = args.data_path
    model_path = args.model_path
    learning_rate = args.learning_rate
    batch_size = args.batch_size
   
    train(data_path, model_path, learning_rate, batch_size)