python_验证曲线_查看不通超参数对模型性能的影响

python_验证曲线_查看不通超参数对模型性能的影响

# 可视化超参数值的效果
# 了解不通超参数对模型性能的影响
import matplotlib.pyplot as plt
import numpy as np
from sklearn.datasets import load_digits
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import validation_curve

# load data
digits = load_digits()

# create feature matrix and target vector
features, target = digits.data, digits.target

# create range of values for parameter
# 创建超参数的变化范围
param_range = np.arange(