LightGBM — это эффективный градиентный бустинг фреймворк, разработанный компанией Microsoft. Он предназначен для работы с большими объемами данных и создания моделей машинного обучения, которые могут решать задачи классификации и регрессии. LightGBM использует уникальные алгоритмы, такие как Gradient-Based One-Side Sampling (GOSS) и Exclusive Feature Bundling (EFB), чтобы ускорить обучение модели и улучшить ее точность.
pip install lightgbm
import lightgbm as lgbfrom sklearn.datasets import load_bostonfrom sklearn.model_selection import train_test_split
boston = load_boston()X, y = boston.data, boston.targetX_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
train_data = lgb.Dataset(X_train, label=y_train)test_data = lgb.Dataset(X_test, label=y_test)params = { 'boosting_type': 'gbdt', 'objective': 'regression', 'metric': {'l2', 'l1'}, 'num_leaves': 31, 'learning_rate': 0.05, 'feature_fraction': 0.9, 'bagging_fraction': 0.8, 'bagging_freq': 5, 'verbose': 0}gbm = lgb.train(params, train_data, num_boost_round=20, valid_sets=test_data, early_stopping_rounds=5)
Следуя этим шагам, вы сможете эффективно использовать LightGBM для построения точных и быстрых моделей машинного обучения.