A simple implementation to regression problems using Python 2.7 and LightGBM. LGBMRegressor is a general purpose script for model training using LightGBM. It contains:
- Functions to preprocess a data file into the necessary train and test Datasets for LightGBM
- Functions to convert categorical variables into dense vectors
- Additional user functionality that allows notification updates to be sent to a user's chosen Slack channel, so that you know when your model has finished training
- Implementation of the DART algorithm
- Implementation of early stopping cross-validation to find the best tree
Follow instructions here
Here are some additional resources if you are looking to explore LightGBM more extensively:
- LightGBM Features
- LightGBM's Python API
- LightGBM's complete parameter specification
- DART: Dropouts meet Multiple Additive Regression Trees
- Optimisation in network communication
- Voting parallel learning
- Fair loss
LightGBM is a relatively new GBM framework. If you are looking to learn more about GBMs in general, here are some additional resources about XGBoost, which has more extensive documentation and is configured very similarly to LightGBM:
- XGBRegressor is a general purpose script for model training using XGBoost
- Introduction to Boosted Trees and the XGBoost algorithm
- The Python API documentation for XGBoost
- Complete Guide to Parameter Tuning in XGBoost
- Tong He's XGBoost presentation