我正在尝试使用Random Forests在Python中执行聚类。在随机森林的R实现中,您可以设置一个标记来获取邻近矩阵。我似乎无法在随机森林的python scikit版本中找到类似的东西。有谁知道python版本是否有相同的计算?
答案 0 :(得分:15)
我们尚未在Scikit-Learn(尚未)中实现邻近矩阵。
但是,这可以通过依赖我们的决策树实现中提供的apply
函数来完成。也就是说,对于数据集中的所有样本对,迭代森林中的决策树(通过forest.estimators_
)并计算它们落入同一叶子的次数,即次数{{1为该对中的两个样本提供相同的节点ID。
希望这会有所帮助。
答案 1 :(得分:3)
基于Gilles Louppe回答我写了一个函数。我不知道它是否有效,但它有效。最好的问候。
def proximityMatrix(model, X, normalize=True):
terminals = model.apply(X)
nTrees = terminals.shape[1]
a = terminals[:,0]
proxMat = 1*np.equal.outer(a, a)
for i in range(1, nTrees):
a = terminals[:,i]
proxMat += 1*np.equal.outer(a, a)
if normalize:
proxMat = proxMat / nTrees
return proxMat
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_breast_cancer
train = load_breast_cancer()
model = RandomForestClassifier(n_estimators=500, max_features=2, min_samples_leaf=40)
model.fit(train.data, train.target)
proximityMatrix(model, train.data, normalize=True)
## array([[ 1. , 0.414, 0.77 , ..., 0.146, 0.79 , 0.002],
## [ 0.414, 1. , 0.362, ..., 0.334, 0.296, 0.008],
## [ 0.77 , 0.362, 1. , ..., 0.218, 0.856, 0. ],
## ...,
## [ 0.146, 0.334, 0.218, ..., 1. , 0.21 , 0.028],
## [ 0.79 , 0.296, 0.856, ..., 0.21 , 1. , 0. ],
## [ 0.002, 0.008, 0. , ..., 0.028, 0. , 1. ]])
答案 2 :(得分:0)
目前在python中没有实现此功能。我第一次尝试here。如果有人有兴趣将这些方法添加到scikit中,那就太棒了。