from sklearn.metrics.pairwise import cosine_similarity a = [[1, 3, 2], [2, 2, 1]] print(cosine_similarity(a))