绘制随机森林每棵树的决策边界
首先导入必要的库函数:
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_moons
from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt
import numpy as np
import mglearn
使用make_moons数据集构造一个有5棵树的随机森林模型,观察每棵树的决策边界。
X, y = make_moons(n_samples=100,noise=0.25,random_state=3)
X_train, X_test, y_train, y_test = train_test_split(X, y, stratify=y, random_state=42)
forest = RandomForestClassifier(n_estimators=5,random_state=2)
forest