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