40
правок
Изменения
→Пример использования с помощью библиотеки xgboost
==Пример использования с помощью библиотеки xgboost==
Загрузка датасета.
from sklearn import datasets
iris = datasets.load_iris()
y = iris.target
Разделение датасета на обучающую/тестовую выборку.
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
Импорт ''XGBoost'' и создание необходимых объектов.
import xgboost as xgb
dtrain = xgb.DMatrix(X_train, label=y_train)
dtest = xgb.DMatrix(X_test, label=y_test)
Задание параметров модели.
param = {
'max_depth': 3,
num_round = 20
Обучение.
bst = xgb.train(param, dtrain, num_round)
preds = bst.predict(dtest)
Определение качества модели на тестовой выборке.
import numpy as np
from sklearn.metrics import precision_score
