【发布时间】:2021-09-20 03:49:15
【问题描述】:
-
我发现了 299 篇文章之间的相似性,但我需要一些帮助来创建包含 20-30 个最相似文档的热图。
-
我尝试使用 sns.heatmap,但无法绘制它。我认为问题在于试图创建包含所有 299 个文档的热图。
-
我该怎么办?请给点意见
from sklearn.metrics.pairwise import cosine_similarity arr = cosine_similarity(tfidf_matrix_train, tfidf_matrix_train) print(arr)
输出:
df_similar = pd.DataFrame(arr)
输出:
【问题讨论】:
标签: python dataframe heatmap data-mining cosine-similarity