По примеру с:
(https://scikit -learn.org / стабильный / auto_examples / ансамбль / plot_ensemble_oob.html # sphx-GLR-авто-примеры-ансамблю сюжетно-ансамблю OOB-ру )
Я хочу построить изображение OOBerror-n_estimator (мой код приведен ниже), но результат сбит с толку (он растет !!!!) ——
введите описание изображения здесь
Я не понимаю причину, кто-нибудь может знать это?
PS: мои данные не сбалансированы, поэтому я установил «class_weight», а также установил «начальную загрузку» randomforest ...
rf = RandomForestClassifier(n_estimators=100,
warm_start=True,
class_weight = "balanced_subsample",
bootstrap = True,
oob_score = True,
random_state = 123
)
# Map a classifier name to a list of (<n_estimators>, <error rate>) pairs.
error_rate = []
n_estimat=[]
# Range of `n_estimators` values to explore.
min_estimators = 1
max_estimators = 100
for i in range(min_estimators, max_estimators + 1):
rf.set_params(n_estimators=i)
rf.fit(x_train, y_train)
# Record the OOB error for each `n_estimators=i` setting.
oob_error = 1 - rf.oob_score_
error_rate.append(oob_error)
n_estimat.append(i)