sklearn.model_selection.validation_curve 验证曲线
验证曲线是用来提高模型的性能,验证曲线和学习曲线很相近,不同的是这里画出的是不同参数下模型的准确率而不是不同训练集大小下的准确率,主要用来调参,validation_curve方法使用采样k折交叉验证来评估模型的性能。
sklearn.model_selection.validation_curve(estimator, X, y, *, param_name, param_range, groups=None, cv=None, scoring=None, n_jobs=None, pre_dispatch='all', verbose=0, error_score=nan)
参数:
param_name :str,要评估的参数值,如果当model为SVC时,改变gamma的值,求最好的那个gamma值
param_range:参数的范围
返回:
train_scores:训练集得分
test_scores:测试集得分
from sklearn import datasets from sklearn.ensemble import RandomForestClassifier from sklearn.model_selection import validation_curve import numpy as np import matplotlib.pyplot as plt (X,y) = datasets.load_digits(return_X_y=True) # print(X[:2,:]) param_range = [10,20,40,80,160,250] train_score,test_score = validation_curve(RandomForestClassifier(),X,y,param_name='n_estimators',param_range=param_range,cv=10,scoring='accuracy') train_score = np.mean(train_score,axis=1) test_score = np.mean(test_score,axis=1) plt.plot(param_range,train_score,'o-',color = 'r',label = 'training') plt.plot(param_range,test_score,'o-',color = 'g',label = 'testing') plt.legend(loc='best') plt.xlabel('number of tree') plt.ylabel('accuracy') plt.show()
分类:
sklearn
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· winform 绘制太阳,地球,月球 运作规律
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人