import argparse
import ast


def define_parser():
    parser = argparse.ArgumentParser(description='Classification by BiLSTM')
    parser.add_argument('--seq',
                        required=False,
                        metavar=False,
                        help='Use sequence model or single model.',
                        type=ast.literal_eval,
                        dest='seq')
    parser.add_argument('--bs',
                        required=False,
                        metavar=32,
                        help="Batch Size",
                        type=int,
                        dest='bs')
    parser.add_argument('--ep',
                        required=False,
                        metavar=300,
                        help="Epoch",
                        type=int,
                        dest='ep')
    parser.add_argument('--tm',
                        required=False,
                        metavar=0,
                        help="Training Mode: 0 标准模式 1 GAP模式 2 BiLSTM模式",
                        type=int,
                        dest='tm')
    parser.add_argument('--net',
                        required=False,
                        metavar='resnet50',
                        help="Select a base model.",
                        type=str,
                        dest='net')
    args = parser.parse_args()
    return args
