Skip to content

Commit

Permalink
reduce params
Browse files Browse the repository at this point in the history
  • Loading branch information
PondiB committed Dec 11, 2023
1 parent b4068d6 commit b83aaad
Showing 1 changed file with 3 additions and 43 deletions.
46 changes: 3 additions & 43 deletions proposals/ml_fit_class_xgboost.json
Original file line number Diff line number Diff line change
@@ -1,15 +1,15 @@
{
"id": "ml_fit_class_xgboost",
"summary": "Train an XGBoost classification model",
"description": "Executes the fit of an XGBoost classification model based on training data.",
"description": "Fit an XGBoost classification model to training data. XGBoost is a high-performance, flexible, and portable distributed gradient boosting library. It implements machine learning algorithms within the Gradient Boosting framework, featuring parallel tree boosting for efficiency",
"categories": [
"machine learning"
],
"experimental": true,
"parameters": [
{
"name": "predictors",
"description": "The predictors for the XGBoost classification model as a vector data cube. Aggregated to the features (vectors) of the target input variable.",
"description": "The predictors for the XGBoost classification model as a vector data cube. They are the independent variables that the XGBoost algorithm analyses to learn patterns and relationships within the data.",
"schema": {
"type": "object",
"subtype": "datacube",
Expand All @@ -25,7 +25,7 @@
},
{
"name": "target",
"description": "Labeled data for XGBoost classification, aligning with predictor values based on a shared geometry dimension. This ensures a clear connection between predictor rows and labels, allowing the model to associate specific predictor values with rows during training.",
"description": "Labeled data for XGBoost classification, aligning with predictor values based on a shared geometry dimension. This ensures a clear connection between predictor rows and labels.",
"schema": {
"type": "object",
"subtype": "datacube",
Expand Down Expand Up @@ -84,46 +84,6 @@
"minimum": 0
}
},
{
"name": "max_delta_step",
"description": "Maximum delta step we allow each tree's weight estimation to be.",
"optional": true,
"default": 1,
"schema": {
"type": "number",
"minimum": 0
}
},
{
"name": "nfold",
"description": "Number of folds for cross-validation.",
"optional": true,
"default": 5,
"schema": {
"type": "integer",
"minimum": 2
}
},
{
"name": "nrounds",
"description": "Number of boosting rounds.",
"optional": true,
"default": 100,
"schema": {
"type": "integer",
"minimum": 1
}
},
{
"name": "early_stopping_rounds",
"description": "Activates early stopping. Validation metric needs to improve at least once in every early_stopping_rounds round(s) to continue training.",
"optional": true,
"default": 20,
"schema": {
"type": "integer",
"minimum": 1
}
},
{
"name": "seed",
"description": "A randomization seed to use for the random sampling in training. If not given or `null`, no seed is used and results may differ on subsequent use.",
Expand Down

0 comments on commit b83aaad

Please sign in to comment.