mirror of
https://github.com/clearml/clearml-docs
synced 2025-02-01 06:56:56 +00:00
1.6 KiB
1.6 KiB
title |
---|
XGBoost and scikit-learn |
The xgboost_sample.py example demonstrates integrating ClearML into code that uses XGBoost.
The example does the following:
- Trains a network on the scikit-learn iris classification dataset using XGBoost
- Scores accuracy using scikit-learn
- ClearML automatically logs the input model registered by XGBoost, and the output model (and its checkpoints), feature importance plot, and tree plot created with XGBoost.
- Creates an experiment named
XGBoost simple example
, which is associated with theexamples
project.
Plots
The feature importance plot and tree plot appear in the project's page in the ClearML web UI, under PLOTS.
Console
All other console output appear in CONSOLE.
Artifacts
Models created by the experiment appear in the experiment’s ARTIFACTS tab. ClearML automatically logs and tracks models and any snapshots created using XGBoost.
Clicking on the model's name takes you to the model’s page, where you can view the model’s details and access the model.