Soft Cosine Similarity Python