Snowball's picture
Upload 6 files
f115ecd
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