Optuna + LightGBM \= OptGBM
Project description
OptGBM
OptGBM (= Optuna + LightGBM) provides a scikit-learn compatible estimator that tunes hyperparameters in LightGBM with Optuna.
Examples
from optgbm.sklearn import OGBMRegressor
from sklearn.datasets import load_boston
reg = OGBMRegressor(random_state=0)
X, y = load_boston(return_X_y=True)
reg.fit(X, y)
score = reg.score(X, y)
Installation
pip install optgbm
Testing
python setup.py test
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
OptGBM-0.5.0.tar.gz
(11.3 kB
view hashes)
Built Distribution
OptGBM-0.5.0-py3-none-any.whl
(9.2 kB
view hashes)