sklearn.ensemble.RandomForestClassifier中的邻近矩阵

时间:2013-09-09 16:49:38

标签: python scikit-learn random-forest

我正在尝试使用Random Forests在Python中执行聚类。在随机森林的R实现中,您可以设置一个标记来获取邻近矩阵。我似乎无法在随机森林的python scikit版本中找到类似的东西。有谁知道python版本是否有相同的计算?

3 个答案:

答案 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中,那就太棒了。