Xgboost Regressor
xgboost_regressorTrain a regularised gradient-boosting regression model on a prepared numeric split to get train/test R2, MAE, MSE, RMSE, MAPE, top features, and warnings.
Instructions
Train an XGBoost regressor: fast, regularised gradient boosting. Requires the optional xgboost package. The split must have only numerical, complete features (no scaling is needed). Args: split_id (str): ID of a split made by train_test_split. n_estimators (int): Number of boosting rounds. Default is 100. learning_rate (float): Shrinkage per round, greater than 0. Default is 0.1. max_depth (int): Depth of each tree. Default is 6. subsample (float): Fraction of rows used per tree, in (0, 1]. Default is 1.0. colsample_bytree (float): Fraction of features used per tree, in (0, 1]. Default is 1.0. random_state (int): Seed. Default is 42. Returns: dict: model_id, train/test metrics (R2, MAE, MSE, RMSE, MAPE), top features and warnings.
Tool call template
Replace each placeholder with a value matching the input schema. Only required parameters are included; add optional parameters as needed.
{
"id": 1,
"jsonrpc": "2.0",
"method": "tools/call",
"params": {
"arguments": {
"split_id": "<split_id>"
},
"name": "xgboost_regressor"
}
}Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| split_id | Yes | ||
| max_depth | No | ||
| subsample | No | ||
| n_estimators | No | ||
| random_state | No | ||
| learning_rate | No | ||
| colsample_bytree | No |