File size: 688 Bytes
f115ecd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import argparse
# TODO: add help for the parameters
def get_parser_main_model():
parser = argparse.ArgumentParser()
# TODO: basic parameters training related
# for embed
parser.add_argument('--language', type=str, default='English', help='text language')
parser.add_argument('--mode', type=str, choices=['embed', 'fast_detect', 'precise_detect'], default='embed', help='Mode options: embed (default), fast_detect, precise_detect')
parser.add_argument('--tau_word', type=float, default=0.8, help='word-level similarity thresh')
parser.add_argument('--lamda', type=float, default=0.83, help='word-level similarity weight')
return parser
|