亲和传播拟合、拟合预测和预测方法

数据挖掘 Python 数据挖掘 scikit-学习
2022-03-08 16:38:45

我正在使用 sklearn 亲和传播算法，如下所示。

affprop = sklearn.cluster.AffinityPropagation(affinity="precomputed", damping=0.5)

我还为我正在使用的数据创建了一个相似度矩阵。现在我想使用我的相似度矩阵在亲和力传播模型中使用。

在sklearn 中，他们有不同的方法，例如 fit、fit_predict、predict。所以，我不确定要使用什么。

如果我使用它是否正确，

affprop.fit(my similarity matrix)
1个回答

按照您的指示使用是正确的。

例如，在下面，我使用 levenshtein 来制作相似度矩阵。然后通过亲和力算法将此矩阵用于聚类。

lev_similarity = -1*np.array([[distance.levenshtein(w1,w2) for w1 in words]
                              for w2 in words])

af = affprop.fit(lev_similarity)
其它你可能感兴趣的问题
上一篇机器学习预测网页上的 HTML 元素？ 下一篇如何获得复合词的语义词嵌入？