Představení XGBoost

Extreme Gradient Boosting with XGBoost

Sergey Fogelson

Head of Data Science, TelevisaUnivision

Co je XGBoost?

Optimalizovaná knihovna pro gradient boosting
Původně napsána v C++
Dostupná API v několika jazycích:
- Python
- R
- Scala
- Julia
- Java

Proč je XGBoost tak populární?

Rychlost a výkon
Jádrový algoritmus je paralelizovatelný
Konzistentně překonává jednoalgoritmické metody
Špičkový výkon v mnoha úlohách strojového učení

Použití XGBoost: rychlý příklad

import xgboost as xgb
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split

class_data = pd.read_csv("classification_data.csv")

X, y = class_data.iloc[:,:-1], class_data.iloc[:,-1]

X_train, X_test, y_train, y_test= train_test_split(X, y, 
        test_size=0.2, random_state=123)

xg_cl = xgb.XGBClassifier(objective='binary:logistic', 
        n_estimators=10, seed=123)

xg_cl.fit(X_train, y_train)

preds = xg_cl.predict(X_test)

accuracy = float(np.sum(preds==y_test))/y_test.shape[0]

print("accuracy: %f" % (accuracy))

accuracy: 0.78333

Začněme používat XGBoost!

Extreme Gradient Boosting with XGBoost