mlflow--mlflow/examples/lightgbm/lightgbm_sklearn

XGBoost Scikit-learn Model Example

This example trains an LightGBM.LGBMClassifier with the diabetes dataset and logs hyperparameters, metrics, and trained model.

Like the other LightGBM example, we enable autologging for LightGBM scikit-learn models via mlflow.lightgbm.autolog(). Saving / loading models also supports LightGBM scikit-learn models.

You can run this example using the following command:

python train.py