test_simbert.py 440 B

1234567891011121314
  1. from nlpcda import Simbert
  2. simbert_config = {
  3. 'model_path': './classify_service/chinese_roformer-sim-char_L-12_H-768_A-12',
  4. 'CUDA_VISIBLE_DEVICES': '0,1',
  5. 'max_len': 64,
  6. 'seed': 1
  7. }
  8. if __name__ == '__main__':
  9. l = ["登录", "注册", "注销", "点击", "关闭"]
  10. for item in l:
  11. simbert = Simbert(config=simbert_config)
  12. synonyms = simbert.replace(sent=l, create_num=5)
  13. print(item, synonyms)