java 获取叶子节点_获取叶子节点（随机森林）的决策路径中的所有功能

我只使用2个决策树在虹膜数据集上有一个简单的随机森林分类器的示例代码 . 此代码最好在jupyter笔记本中运行 .

# Setup

%matplotlib inline

import matplotlib.pyplot as plt

from sklearn.datasets import load_iris

from sklearn.cross_validation import train_test_split

from sklearn.ensemble import RandomForestClassifier

from sklearn.metrics import confusion_matrix

import numpy as np

# Set seed for reproducibility

np.random.seed(1015)

# Load the iris data

iris = load_iris()

# Create the train-test datasets

X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target)

np.random.seed(1039)

# Just fit a simple random forest classifier with 2 decision trees

rf = RandomForestClassifier(n_estimators = 2)

rf.fit(X = X_train, y = y_train)

# Define a function to draw the decision trees in IPython

# Adapte