机器学习中如何使参数最优

最新推荐文章于 2023-10-01 15:49:56 发布

可怜又无助的迪迪迪

最新推荐文章于 2023-10-01 15:49:56 发布

阅读量1.3k

点赞数

分类专栏：机器学习sklearn 文章标签：机器学习随机森林

本文链接：https://blog.csdn.net/z1139269312/article/details/121776004

版权

学习曲线

from sklearn.datasets import load_breast_cancer
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import GridSearchCV
from sklearn.model_selection import cross_val_score
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

data = load_breast_cancer()
rfc = RandomForestClassifier(n_estimators=100,random_state=90)
score_pre = cross_val_score(rfc,data.data,data.target,cv=10,scoring='accuracy').mean()
print(score_pre)

scorel = []
for i in range(0,200,10):
    rfc = RandomForestClassifier(n_estimators=i+1,n_jobs=-1,random_state=90)
    score = cross_val_score(rfc,data.data,data.target,cv=10).mean()
    scorel.append(score)
print(max(scorel),(scorel.index(max(sc