| # we disable them for now. | |
| # import argparse | |
| # from idiomify.fetchers import fetch_config, fetch_rd, fetch_idioms | |
| # from transformers import BertTokenizer | |
| # from termcolor import colored | |
| # | |
| # | |
| # def main(): | |
| # parser = argparse.ArgumentParser() | |
| # parser.add_argument("--model", type=str, | |
| # default="alpha") | |
| # parser.add_argument("--ver", type=str, | |
| # default="eng2eng") | |
| # parser.add_argument("--sent", type=str, | |
| # default="to avoid getting to the point") | |
| # args = parser.parse_args() | |
| # config = fetch_config()[args.model][args.ver] | |
| # config.update(vars(args)) | |
| # idioms = fetch_idioms(config['idioms_ver']) | |
| # rd = fetch_rd(config['model'], config['ver']) | |
| # rd.eval() | |
| # tokenizer = BertTokenizer.from_pretrained(config['bert']) | |
| # X = T.inputs([("", config['sent'])], tokenizer, config['k']) | |
| # probs = rd.P_wisdom(X).squeeze().tolist() | |
| # wisdom2prob = [ | |
| # (wisdom, prob) | |
| # for wisdom, prob in zip(idioms, probs) | |
| # ] | |
| # # sort and append | |
| # res = list(sorted(wisdom2prob, key=lambda x: x[1], reverse=True)) | |
| # print(f"query: {colored(text=config['sent'], color='blue')}") | |
| # for idx, (idiom, prob) in enumerate(res): | |
| # print(idx, idiom, prob) | |
| # | |
| # | |
| # if __name__ == '__main__': | |
| # main() | |