Back to Mlflow

XGBoost Scikit-learn Model Example

examples/xgboost/xgboost_sklearn/README.md

3.12.0517 B
Original Source

XGBoost Scikit-learn Model Example

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

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

You can run this example using the following command:

python train.py