parse_args()

import argparse

def parse_args():
    parser = argparse.ArgumentParser(description='DeepRec')
    parser.add_argument('--num_epochs', type=int, default=100)
    parser.add_argument('--emb_size', type=int, default=64)
    parser.add_argument('--len_Seq', type=int, default=5)
    parser.add_argument('--len_Tag', type=int, default=3)
    parser.add_argument('--len_Pred', type=int, default=1)
    parser.add_argument('--neg_sample', type=int, default=1)
    parser.add_argument('--batch_size', type=int, default=1024)
    parser.add_argument('--v_size', type=int, default=4)
    parser.add_argument('--learning_rate', type=float, default=1e-2)
    parser.add_argument('--keep_prob', type=float, default=1.0)
    parser.add_argument('--h_size', type=int, default=16)
    parser.add_argument('--l2_lambda', type=float, default=1e-6)
    return parser.parse_args()

# 使用
args = parse_args()
len_Seq = args.len_Seq  # 序列的长度
len_Tag = args.len_Tag  # 训练时目标的长度
len_Pred = args.len_Pred  # 预测时目标的长度
......

你可能感兴趣的:(python)