helper.py

# -*- coding:utf-8 -*-
'''
@Author: yanwii
@Date: 2019-02-26 10:48:20
'''

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("-m", "--mode", type=str, help="google bert pretrained model", default="normal")
parser.add_argument("-dd", "--data_dir", type=str, help="tran, dev and test data dir", default="data/")
parser.add_argument("-bc", "--bert_config", type=str, help="bert config file dir", default="bert_model/bert_config.json")
parser.add_argument("-od", "--output_dir", type=str, help="output dir", default="output")
parser.add_argument("-ic", "--init_checkpoint", type=str, help="bert model dir", default="bert_model/")
parser.add_argument("-v", "--vocab_dir", type=str, help="vocab dir", default="bert_model/vocab.txt")
parser.add_argument("-e", "--entry", type=str, default="train")
parser.add_argument("-lr", "--learning_rate", type=float, default=5e-5, help="learning rate")
ARGS = parser.parse_args()