Tracking XGBoost Training

sigopt.com Log In / Sign Up
Search…
Welcome to SigOpt!
Quick Start Tutorials
Tracking Your Training Runs
Optimization Experiments
XGBoost Integration
Installation
Tracking XGBoost Training
Tuning XGBoost Models
HyperOpt
Advanced Experimentation
Tracking XGBoost Training
sigopt.xgboost.run
This function provides a wrapper around the xgboost.train API ; it automatically tracks and records all aspects of the model training process to SigOpt. The sigopt.xgboost.run API closely matches the xgboost.train API to minimize any disruption to your workflow, while allowing you to enjoy all the features provided by the SigOpt experiment management platform.
Example
Before executing the example code, make sure you have set up your API token and project name as environment variables. You can follow up the set up instructions here or set them in Python like the example below.
1
import os
2
os.environ["SIGOPT_API_TOKEN"] = # PASTE API TOKEN FROM https://app.sigopt.com/tokens/info
3
os.environ["SIGOPT_PROJECT"] = "my-xgboost-project"
Copied!
Now we demonstrate how to modify your existing XGBoost training script using the xgboost.train API to our sigopt.xgboost.run to enable automatic tracking with SigOpt.
Existing code with xgboost.train API
1
from sklearn import datasets
2
from sklearn.model_selection import train_test_split
3
import xgboost
4
​
5
bc = datasets.load_breast_cancer()
6
(Xtrain, Xtest, ytrain, ytest) = train_test_split(bc.data, bc.target, test_size=0.5, random_state=42)
7
dtrain = xgboost.DMatrix(data=Xtrain, label=ytrain, feature_names=bc.feature_names)
8
dtest = xgboost.DMatrix(data=Xtest, label=ytest, feature_names=bc.feature_names)
9
​
10
params = {
11
  "eta": 0.05,
12
  "gamma": 1,
13
  "max_depth": 4,
14
  "objective": "binary:logistic"
15
}
16
booster = xgboost.train(
17
  params,
18
  dtrain,
19
  num_boost_round=25,
20
  evals=[(dtest, "val_set")],
21
)
Copied!
Updated with sigopt.xgboost.run API
1
from sklearn import datasets
2
from sklearn.model_selection import train_test_split
3
import xgboost
4
import sigopt.xgboost
5
​
6
bc = datasets.load_breast_cancer()
7
(Xtrain, Xtest, ytrain, ytest) = train_test_split(bc.data, bc.target, test_size=0.5, random_state=42)
8
dtrain = xgboost.DMatrix(data=Xtrain, label=ytrain, feature_names=bc.feature_names)
9
dtest = xgboost.DMatrix(data=Xtest, label=ytest, feature_names=bc.feature_names)
10
​
11
params = {
12
  "eta": 0.05,
13
  "gamma": 1,
14
  "max_depth": 4,
15
  "objective": "binary:logistic"
16
}
17
run_options = {
18
 "name": "xgboost-tracking-with-sigopt"
19
}
20
xgb_run = sigopt.xgboost.run(
21
  params,
22
  dtrain,
23
  num_boost_round=25,
24
  evals=[(dtest,"val_set")],
25
  run_options=run_options,
26
)
27
booster = xgb_run.model
28
print(f"View run at https://app.sigopt.com/run/{xgb_run.run.id}")
Copied!
All relevant information from the XGBoost model training process is automatically recorded in a SigOpt Run and can viewed on the Run page in the web app.
Automated Logging with sigopt.xgboost.run
sigopt.xgboost.run automatically logs the relevant information and metadata of the XGBoost training process and the resulting model. We list all the automated logging functionalities below.
Logging Parameters
sigopt.xgboost.run automatically logs all XGBoost parameters that you specify via the params argument. These values are displayed under the Parameter Values section on the Run page. In additional, sigopt.xgboost.run also records the XGBoost default values for all other parameters.
Logging Metrics
If the evaluation metrics are specified in the params argument via eval_metric, sigopt.xgboost.run automatically logs every evaluation metric value for each evaluation dataset after the training has completed. The metric values are listed under Metrics on the Run page.
1
sigopt.xgboost.run(
2
  params={
3
    "objective": "binary:logistic", # SigOpt logs additional metrics for binary classification task
4
    "eval_metric": ["aucpr", "logloss"], # Automatically logged to SigOpt
5
  },
6
  dtrain=dtrain,
7
  evals=[(dtest, "val_set")],
8
)
Copied!
For certain learning tasks, sigopt.xgboost.run logs additional metrics computed using the evaluation datasets. The supported tasks are inferred from the objective parameter prefix. We list the objective prefixes and the corresponding metrics below.
Metric Name
Objective Prefix
Description
accuracy
binary:, multi:
Percentage of predicted labels that match true labels.
precision
binary:, multi:
Ratio of true positive predictions to all positive predictions.
recall
binary:, multi:
Ratio of true positive predictions to all positive labels.
F1
binary:, multi:
The harmonic mean of precision and recall.
mean squared error
reg:
Squared error between each predicted label and true label.
mean absolute error
reg:
Absolute error between each predicted label and true label.
Training Time
All objectives
Number of seconds to execute xgboost.train.
Logging Checkpoints
If the evaluation metrics are specified in the params, sigopt.xgboost.run also automatically tracks the iterative evaluation metric using checkpoints. You can control the frequency of the checkpoint logging using the verbose_eval argument. If verbose_eval is an integer, then SigOpt logs the evaluation metric every given verbose_eval boosting rounds. If verbose_eval is True, then SigOpt logs the evaluation metric every boosting round. If it is False, SigOpt logs the evaluation metric every 5 boosting rounds.
Caveat: SigOpt only supports up to 200 total checkpoints per Run. sigopt.xgboost.run will automatically adjust the checkpoint logging frequency accordingly to guarantee that the total number of checkpoints does not exceed this limit.
Logging Feature Importances
sigopt.xgboost.run automatically computes and tracks the feature importances of the training dataset. Generally, feature importances indicate how useful each feature is in determining boosted decision tree predictions. Feature importances are computed using the XGBoost get_score function with importance_type=weight. The top 50 features are tracked and stored on the Run page.
Logging Metadata
sigopt.xgboost.run automatically tracks additional metadata associated with the model training process such as dataset dimensions, dataset names, XGBoost version, and more.
Input Arguments for sigopt.xgboost.run
All arguments supported by xgboost.train are supported by sigopt.xgboost.run. You can preserve your existing XGBoost code and realize the benefits of tracking the model training as a SigOpt TrainingRun with minimal code change. To better understand the XGBoost Learning API that this integration is based on, you can refer to the XGBoost documentation pages on the Training API and XGBoost Parameters respectively.
Optional Argument - run_options
You can control what aspect of the model training to be logged by the sigopt.xgboost.run call with the run_options argument. Here, we provide an example of using run_options argument to create a Run with an assigned name and selectively turn off checkpoint logging and stdout logging.
1
run_options = {
2
  "autolog_checkpoints": False, # Turn off checkpoint logging
3
  "autolog_stdout": False, # Turn off stdout logging
4
  "name": "custom-logging-run",
5
}
6
​
7
xgb_run = sigopt.xgboost.run(
8
  params={"eta": 0.02, "objective": "reg:squarederror"}
9
  dtrain=dtrain,
10
  num_boost_round=23,
11
  evals=[(dtest, "testset")],
12
  run_options=run_options, # Passing in the run_options
13
)
Copied!
The following table lists the automated logging capabilities you can control using the run_options dict.
Key
Type
Default value
Description
autolog_checkpoints
Boolean
True
Automatically log the iterated evaluation metrics for every specified interval.
autolog_feature_importances
Boolean
True
Automatically log the dataset feature importance scores.
autolog_metrics
Boolean
True
Automatically log additional evaluation metrics.
autolog_xgboost_defaults
Boolean
True
Automatically log XGBoost default values for unspecified parameters.
autolog_stderr
Boolean
True
Automatically log stderr from the xgboost.train call.
autolog_stdout
Boolean
True
Automatically log stdout from the xgboost.train call.
autolog_sys_info
Boolean
True
Automatically log the Python version and XGBoost version.
name
string
None
Assign a name to a newly created TrainingRun.
run
​RunContext​
None
Pass an existing RunContext to sigopt.xgboost.run. This is useful for creating a custom Experiment loop.
Output of sigopt.xgboost.run
The sigopt.xgboost.run function returns an XGBRun object which allows you to access the resulting XGBoost Booster object via .model and the SigOpt RunContext object via .run. In the following example, we demonstrate how to continue logging information with the XGBRun object post model training.
1
# save the model and log the filepath after training
2
​
3
xgb_run = sigopt.xgboost.run(params, dtrain)
4
filepath = "./saved_models/xgb_booster.json"
5
xgb_run.model.save_model(filepath) # Saving booster model 
6
xgb_run.run.log_metadata("Booster saved filepath", filepath) # Logging the filepath
7
xgb_run.run.end() # mark run as completed
Copied!