nlp_task.py 682 B

123456789101112131415161718192021
  1. """
  2. @author: luojunhui
  3. """
  4. async def get_nlp_similarity_score(nlp, function, data, use_cache):
  5. """
  6. 获取nlp的相似度分数
  7. """
  8. match function:
  9. case "similarities":
  10. return nlp.base_string_similarity(text_dict=data, use_cache=use_cache)
  11. case "similarities_cross":
  12. return nlp.base_list_similarity(pair_list_dict=data, use_cache=use_cache)
  13. case "similarities_cross_max":
  14. return nlp.max_cross_similarity(data=data)
  15. case "similarities_cross_avg":
  16. return nlp.avg_cross_similarity(data=data)
  17. case "similarities_cross_mean":
  18. return nlp.mean_cross_similarity(data=data)