fork(1) download
  1. import sklearn
  2. from sklearn.feature_extraction.text import TfidfVectorizer
  3. from sklearn.metrics.pairwise import cosine_similarity
  4.  
  5. text1 = 'learn python'
  6. text2 = 'tutorial python'
  7.  
  8. vectorizer = TfidfVectorizer()
  9. vectors = vectorizer.fit_transform([text1,text2])
  10.  
  11. similarity = cosine_similarity(vectors)
  12. print(similarity)
Success #stdin #stdout 0.51s 73428KB
stdin
Standard input is empty
stdout
[[1.         0.33609693]
 [0.33609693 1.        ]]