options.py

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