Примечание
Перейти в конец, чтобы скачать полный код примера или запустить этот пример в браузере с помощью JupyterLite или Binder.
Построение индивидуальных прогнозов регрессии и прогнозов голосующего регрессора
Голосующий регрессор — это метаоцениватель ансамбля, который обучает несколько базовых регрессоров, каждый на всём наборе данных. Затем он усредняет индивидуальные прогнозы, чтобы получить итоговый прогноз. Для предсказания данных мы используем три разных регрессора: GradientBoostingRegressor, RandomForestRegressor и LinearRegression). Затем эти три регрессора будут использованы в VotingRegressor.
Наконец, для сравнения мы построим прогнозы, сделанные всеми моделями.
Мы будем работать с набором данных о диабете, который содержит 10 признаков, собранных у группы пациентов с диабетом. Целевая переменная — количественная оценка прогрессирования заболевания через год после исходного обследования.
# Authors: The scikit-learn developers
# SPDX-License-Identifier: BSD-3-Clause
import matplotlib.pyplot as plt
from sklearn.datasets import load_diabetes
from sklearn.ensemble import (
GradientBoostingRegressor,
RandomForestRegressor,
VotingRegressor,
)
from sklearn.linear_model import LinearRegression
Обучение классификаторов
Сначала загрузим набор данных о диабете и создадим регрессор с градиентным бустингом, регрессор на основе случайного леса и линейную регрессию. Затем используем эти 3 регрессора для построения голосующего регрессора:
X, y = load_diabetes(return_X_y=True)
# Train classifiers
reg1 = GradientBoostingRegressor(random_state=1)
reg2 = RandomForestRegressor(random_state=1)
reg3 = LinearRegression()
reg1.fit(X, y)
reg2.fit(X, y)
reg3.fit(X, y)
ereg = VotingRegressor([("gb", reg1), ("rf", reg2), ("lr", reg3)])
ereg.fit(X, y)
VotingRegressor(estimators=[('gb', GradientBoostingRegressor(random_state=1)),
('rf', RandomForestRegressor(random_state=1)),
('lr', LinearRegression())])
В среде Jupyter повторно выполните эту ячейку, чтобы отобразить представление HTML, или доверьте выполнение блокноту. На GitHub невозможно отобразить представление HTML; попробуйте загрузить эту страницу с помощью nbviewer.org.
Параметры
Обученные атрибуты
| Имя | Тип | Значение |
|---|---|---|
| estimators_ estimators_: список регрессоров Коллекция обученных подоценивателей, заданных в ``estimators``, за исключением тех, которым присвоено значение 'drop'. | list | [GradientBoost...andom_state=1), RandomForestR...andom_state=1), LinearRegression()] |
| n_features_in_ n_features_in_: int Число признаков, обнаруженных во время :term:`fit`. Определяется, только если базовый регрессор предоставляет такой атрибут после обучения. .. versionadded:: 0.24 | int | 10 |
| named_estimators_ named_estimators_: :class:`~sklearn.utils.Bunch` Атрибут для доступа к любому обученному подоценивателю по имени. .. versionadded:: 0.20 | Bunch | {'gb': Gradie...rRegression()} |
Параметры
Обученные атрибуты
Параметры
Обученные атрибуты
Параметры
Обученные атрибуты
3 признака
| votingregressor_gb |
| votingregressor_rf |
| votingregressor_lr |
Выполнение предсказаний
Теперь мы используем каждый из регрессоров, чтобы получить первые 20 предсказаний.
xt = X[:20] pred1 = reg1.predict(xt) pred2 = reg2.predict(xt) pred3 = reg3.predict(xt) pred4 = ereg.predict(xt)
Построение результатов
Наконец, мы визуализируем 20 прогнозов. Красные звёздочки обозначают средний прогноз, полученный с помощью VotingRegressor.
plt.figure()
plt.plot(pred1, "gd", label="GradientBoostingRegressor")
plt.plot(pred2, "b^", label="RandomForestRegressor")
plt.plot(pred3, "ys", label="LinearRegression")
plt.plot(pred4, "r*", ms=10, label="VotingRegressor")
plt.tick_params(axis="x", which="both", bottom=False, top=False, labelbottom=False)
plt.ylabel("predicted")
plt.xlabel("training samples")
plt.legend(loc="best")
plt.title("Regressor predictions and their average")
plt.show()

Общее время выполнения скрипта: (0 минут 1.052 секунды)
Связанные примеры
© 2007–2026 The scikit-learn developers
Licensed under the 3-clause BSD License.
https://scikit-learn.org/1.9/auto_examples/ensemble/plot_voting_regressor.html