(self, sent, k=5, threhold=0.85)
| 26 | self.model = SynonymsGenerator(config['model_path'], config['max_len'], config['seed']) |
| 27 | |
| 28 | def replace(self, sent, k=5, threhold=0.85): |
| 29 | # 产生n个相似句结果，取相似度大于阈值threhold的里面的前k个 |
| 30 | n = k * 4 |
| 31 | synonyms = self.model.gen_synonyms(text=sent, n=n, k=k, threhold=threhold) |
| 32 | return synonyms |
| 33 | |
| 34 | |
| 35 | if __name__ == '__main__': |
no test coverage detected