args_LQ.py

import argparse


def parse_args(mode="train"):
    parser = argparse.ArgumentParser()

    parser.add_argument("--seed", default=42, type=int, help="seed")
    parser.add_argument("--device", default="cpu", type=str, help="cpu or gpu")

    # -- 데이터 경로 및 파일 이름 설정
    parser.add_argument(
        "--data_dir",
        default="/opt/ml/input/data/",
        type=str,
        help="data directory",
    )
    parser.add_argument(
        "--asset_dir", default="asset/", type=str, help="data directory"
    )
    parser.add_argument(
        "--file_name", default="train_data.csv", type=str, help="train file name"
    )

    # -- 모델의 경로 및 이름, 결과 저장
    parser.add_argument(
        "--model_dir", default="models/", type=str, help="model directory"
    )
    parser.add_argument(
        "--model_name", default="model.pt", type=str, help="model file name"
    )
    parser.add_argument(
        "--output_dir", default="output/", type=str, help="output directory"
    )
    parser.add_argument(
        "--test_file_name", default="test_data.csv", type=str, help="test file name"
    )

    parser.add_argument(
        "--max_seq_len", default=30, type=int, help="max sequence length"
    )
    parser.add_argument("--num_workers", default=4, type=int, help="number of workers")

    # 모델
    parser.add_argument(
        "--hidden_dim", default=300, type=int, help="hidden dimension size"
    )
    parser.add_argument("--n_layers", default=2, type=int, help="number of layers")
    parser.add_argument("--n_heads", default=4, type=int, help="number of heads")
    parser.add_argument("--drop_out", default=0.2, type=float, help="drop out rate")

    # 훈련
    parser.add_argument("--n_epochs", default=30, type=int, help="number of epochs")
    parser.add_argument("--batch_size", default=64, type=int, help="batch size")
    parser.add_argument("--lr", default=0.009668, type=float, help="learning rate")
    parser.add_argument("--clip_grad", default=10, type=int, help="clip grad")
    parser.add_argument("--patience", default=10, type=int, help="for early stopping")

    parser.add_argument(
        "--log_steps", default=50, type=int, help="print log per n steps"
    )

    ### 중요 ###
    parser.add_argument("--model", default="LastQuery", type=str, help="model type")
    parser.add_argument("--optimizer", default="adam", type=str, help="optimizer type")
    parser.add_argument(
        "--scheduler", default="plateau", type=str, help="scheduler type"
    )

    # -- Data split methods : default(user), k-fold, ...
    parser.add_argument(
        "--split_method", default="k-fold", type=str, help="data split strategy"
    )
    parser.add_argument(
        "--n_splits", default=5, type=str, help="number of k-fold splits"
    )

    ### Argumentation 관련 ###

    parser.add_argument(
        "--window", default=True, type=bool, help="Arumentation with stridde window"
    )
    parser.add_argument(
        "--shuffle", default=False, type=bool, help="data shuffle option"
    )
    parser.add_argument("--stride", default=80, type=int)
    parser.add_argument("--shuffle_n", default=2, type=int)

    ### Tfixup 관련 ###
    parser.add_argument("--Tfixup", default=False, type=bool, help="Tfixup")

    args = parser.parse_args()

    # args.stride = args.max_seq_len

    return args