网状:提取R中的SKLearn决策路径

时间:2018-07-30 19:15:15

标签: python r scikit-learn decision-tree reticulate

我正在使用sklearn.ensemble的网状结构,我想提取决策路径。

library(reticulate)

# set up data for sklearn
features = names(iris[1:4]) # extract explanatory variables
iris_mat = data.matrix(iris[ , features])
y = as.factor(iris$Species) # format response variable as factoy

# load sklearn
sk = import("sklearn.ensemble")
clf = sk$RandomForestClassifier(n_jobs=10)
clf = RandomForestClassifier(n_jobs=as.integer(10),
                         random_state=as.integer(0), 
                         n_estimators=as.integer(101)) 
# fit the model
clf.fit = clf$fit(X = iris_mat, y= as.double(y))
clf.decision = clf$decision_path(iris_mat)

我的问题是:如何提取与clf.decision对象一起保存的稀疏决策矩阵? clf.decision是一个列表,typeof(clf.decision [[1]])是包含矩阵的“环境”。

谢谢!

0 个答案:

没有答案