Parameters of H2OXGBoost¶
Affected Classes¶
ai.h2o.sparkling.ml.algos.H2OXGBoost
ai.h2o.sparkling.ml.algos.classification.H2OXGBoostClassifier
ai.h2o.sparkling.ml.algos.regression.H2OXGBoostRegressor
Parameters¶
Each parameter has also a corresponding getter and setter method. (E.g.:
label
->getLabel()
,setLabel(...)
)
- calibrationDataFrame
Calibration frame for Platt Scaling. To enable usage of the data frame, set the parameter calibrateModel to True.
Scala default value:
null
; Python default value:None
- ignoredCols
Names of columns to ignore for training.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- monotoneConstraints
A key must correspond to a feature name and value could be 1 or -1
Scala default value:
Map()
; Python default value:{}
Also available on the trained model.
- aucType
Set default multinomial AUC type. Possible values are
"AUTO"
,"NONE"
,"MACRO_OVR"
,"WEIGHTED_OVR"
,"MACRO_OVO"
,"WEIGHTED_OVO"
.Default value:
"AUTO"
Also available on the trained model.
- backend
Backend. By default (auto), a GPU is used if available. Possible values are
"auto"
,"gpu"
,"cpu"
.Default value:
"auto"
Also available on the trained model.
- booster
Booster type. Possible values are
"gbtree"
,"gblinear"
,"dart"
.Default value:
"gbtree"
Also available on the trained model.
- buildTreeOneNode
Run on one node only; no network overhead but fewer cpus used. Suitable for small datasets.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- calibrateModel
Use Platt Scaling (default) or Isotonic Regression to calculate calibrated class probabilities. Calibration can provide more accurate estimates of class probabilities.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- calibrationMethod
Calibration method to use. Possible values are
"AUTO"
,"PlattScaling"
,"IsotonicRegression"
.Default value:
"AUTO"
Also available on the trained model.
- categoricalEncoding
Encoding scheme for categorical features. Possible values are
"AUTO"
,"OneHotInternal"
,"OneHotExplicit"
,"Enum"
,"Binary"
,"Eigen"
,"LabelEncoder"
,"SortByResponse"
,"EnumLimited"
.Default value:
"AUTO"
Also available on the trained model.
- colSampleByLevel
(same as col_sample_rate) Column sample rate (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- colSampleByNode
Column sample rate per tree node (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- colSampleByTree
(same as col_sample_rate_per_tree) Column sample rate per tree (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- colSampleRate
(same as colsample_bylevel) Column sample rate (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- colSampleRatePerTree
(same as colsample_bytree) Column sample rate per tree (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- columnsToCategorical
List of columns to convert to categorical before modelling
Scala default value:
Array()
; Python default value:[]
- convertInvalidNumbersToNa
If set to ‘true’, the model converts invalid numbers to NA during making predictions.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- convertUnknownCategoricalLevelsToNa
If set to ‘true’, the model converts unknown categorical levels to NA during making predictions.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- customMetricFunc
Reference to custom evaluation function, format: language:keyName=funcName.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- dataFrameSerializer
A full name of a serializer used for serialization and deserialization of Spark DataFrames to a JSON value within NullableDataFrameParam.
Default value:
"ai.h2o.sparkling.utils.JSONDataFrameSerializer"
Also available on the trained model.
- detailedPredictionCol
Column containing additional prediction details, its content depends on the model type.
Default value:
"detailed_prediction"
Also available on the trained model.
- distribution
Distribution function. Possible values are
"AUTO"
,"bernoulli"
,"quasibinomial"
,"modified_huber"
,"multinomial"
,"ordinal"
,"gaussian"
,"poisson"
,"gamma"
,"tweedie"
,"huber"
,"laplace"
,"quantile"
,"fractionalbinomial"
,"negativebinomial"
,"custom"
.Default value:
"AUTO"
Also available on the trained model.
- dmatrixType
Type of DMatrix. For sparse, NAs and 0 are treated equally. Possible values are
"auto"
,"dense"
,"sparse"
.Default value:
"auto"
Also available on the trained model.
- eta
(same as learn_rate) Learning rate (from 0.0 to 1.0).
Default value:
0.3
Also available on the trained model.
- evalMetric
Specification of evaluation metric that will be passed to the native XGBoost backend.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- exportCheckpointsDir
Automatically export generated models to this directory.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- featuresCols
Name of feature columns
Scala default value:
Array()
; Python default value:[]
Also available on the trained model.
- foldAssignment
Cross-validation fold assignment scheme, if fold_column is not specified. The ‘Stratified’ option will stratify the folds based on the response variable, for classification problems. Possible values are
"AUTO"
,"Random"
,"Modulo"
,"Stratified"
.Default value:
"AUTO"
Also available on the trained model.
- foldCol
Column with cross-validation fold index assignment per observation.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- gainsliftBins
Gains/Lift table number of bins. 0 means disabled.. Default value -1 means automatic binning.
Default value:
-1
Also available on the trained model.
- gamma
(same as min_split_improvement) Minimum relative improvement in squared error reduction for a split to happen.
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- gpuId
Which GPU(s) to use. .
Scala default value:
null
; Python default value:None
Also available on the trained model.
- growPolicy
Grow policy - depthwise is standard GBM, lossguide is LightGBM. Possible values are
"depthwise"
,"lossguide"
.Default value:
"depthwise"
Also available on the trained model.
- ignoreConstCols
Ignore constant columns.
Scala default value:
true
; Python default value:True
Also available on the trained model.
- interactionConstraints
A set of allowed column interactions.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- keepBinaryModels
If set to true, all binary models created during execution of the
fit
method will be kept in DKV of H2O-3 cluster.Scala default value:
false
; Python default value:False
- keepCrossValidationFoldAssignment
Whether to keep the cross-validation fold assignment.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- keepCrossValidationModels
Whether to keep the cross-validation models.
Scala default value:
true
; Python default value:True
Also available on the trained model.
- keepCrossValidationPredictions
Whether to keep the predictions of the cross-validation models.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- labelCol
Response variable column.
Default value:
"label"
Also available on the trained model.
- learnRate
(same as eta) Learning rate (from 0.0 to 1.0).
Default value:
0.3
Also available on the trained model.
- maxAbsLeafnodePred
(same as max_delta_step) Maximum absolute value of a leaf node prediction.
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- maxBins
For tree_method=hist only: maximum number of bins.
Default value:
256
Also available on the trained model.
- maxDeltaStep
(same as max_abs_leafnode_pred) Maximum absolute value of a leaf node prediction.
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- maxDepth
Maximum tree depth (0 for unlimited).
Default value:
6
Also available on the trained model.
- maxLeaves
For tree_method=hist only: maximum number of leaves.
Default value:
0
Also available on the trained model.
- maxRuntimeSecs
Maximum allowed runtime in seconds for model training. Use 0 to disable.
Default value:
0.0
Also available on the trained model.
- minChildWeight
(same as min_rows) Fewest allowed (weighted) observations in a leaf.
Default value:
1.0
Also available on the trained model.
- minRows
(same as min_child_weight) Fewest allowed (weighted) observations in a leaf.
Default value:
1.0
Also available on the trained model.
- minSplitImprovement
(same as gamma) Minimum relative improvement in squared error reduction for a split to happen.
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- modelId
Destination id for this model; auto-generated if not specified.
Scala default value:
null
; Python default value:None
- nfolds
Number of folds for K-fold cross-validation (0 to disable or >= 2).
Default value:
0
Also available on the trained model.
- normalizeType
For booster=dart only: normalize_type. Possible values are
"tree"
,"forest"
.Default value:
"tree"
Also available on the trained model.
- nthread
Number of parallel threads that can be used to run XGBoost. Cannot exceed H2O cluster limits (-nthreads parameter). Defaults to maximum available.
Default value:
-1
Also available on the trained model.
- ntrees
(same as n_estimators) Number of trees.
Default value:
50
Also available on the trained model.
- offsetCol
Offset column. This will be added to the combination of columns before applying the link function.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- oneDrop
For booster=dart only: one_drop.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- parallelizeCrossValidation
Allow parallel training of cross-validation models.
Scala default value:
true
; Python default value:True
- predictionCol
Prediction column name
Default value:
"prediction"
Also available on the trained model.
- quietMode
Enable quiet mode.
Scala default value:
true
; Python default value:True
Also available on the trained model.
- rateDrop
For booster=dart only: rate_drop (0..1).
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- regAlpha
L1 regularization.
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- regLambda
L2 regularization.
Scala default value:
1.0f
; Python default value:1.0
Also available on the trained model.
- sampleRate
(same as subsample) Row sample rate per tree (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- sampleType
For booster=dart only: sample_type. Possible values are
"uniform"
,"weighted"
.Default value:
"uniform"
Also available on the trained model.
- saveMatrixDirectory
Directory where to save matrices passed to XGBoost library. Useful for debugging.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- scalePosWeight
Controls the effect of observations with positive labels in relation to the observations with negative labels on gradient calculation. Useful for imbalanced problems.
Scala default value:
1.0f
; Python default value:1.0
Also available on the trained model.
- scoreEachIteration
Whether to score during each iteration of model training.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- scoreEvalMetricOnly
If enabled, score only the evaluation metric. This can make model training faster if scoring is frequent (eg. each iteration).
Scala default value:
false
; Python default value:False
Also available on the trained model.
- scoreTreeInterval
Score the model after every so many trees. Disabled if set to 0.
Default value:
0
Also available on the trained model.
- seed
Seed for pseudo random number generator (if applicable).
Scala default value:
-1L
; Python default value:-1
Also available on the trained model.
- skipDrop
For booster=dart only: skip_drop (0..1).
Scala default value:
0.0f
; Python default value:0.0
Also available on the trained model.
- splitRatio
Accepts values in range [0, 1.0] which determine how large part of dataset is used for training and for validation. For example, 0.8 -> 80% training 20% validation. This parameter is ignored when validationDataFrame is set.
Default value:
1.0
- stoppingMetric
Metric to use for early stopping (AUTO: logloss for classification, deviance for regression and anomaly_score for Isolation Forest). Note that custom and custom_increasing can only be used in GBM and DRF with the Python client. Possible values are
"AUTO"
,"deviance"
,"logloss"
,"MSE"
,"RMSE"
,"MAE"
,"RMSLE"
,"AUC"
,"AUCPR"
,"lift_top_group"
,"misclassification"
,"mean_per_class_error"
,"anomaly_score"
,"AUUC"
,"ATE"
,"ATT"
,"ATC"
,"qini"
,"custom"
,"custom_increasing"
.Default value:
"AUTO"
Also available on the trained model.
- stoppingRounds
Early stopping based on convergence of stopping_metric. Stop if simple moving average of length k of the stopping_metric does not improve for k:=stopping_rounds scoring events (0 to disable).
Default value:
0
Also available on the trained model.
- stoppingTolerance
Relative tolerance for metric-based stopping criterion (stop if relative improvement is not at least this much).
Default value:
0.001
Also available on the trained model.
- subsample
(same as sample_rate) Row sample rate per tree (from 0.0 to 1.0).
Default value:
1.0
Also available on the trained model.
- treeMethod
Tree method. Possible values are
"auto"
,"exact"
,"approx"
,"hist"
.Default value:
"auto"
Also available on the trained model.
- tweediePower
Tweedie power for Tweedie regression, must be between 1 and 2.
Default value:
1.5
Also available on the trained model.
- validationDataFrame
A data frame dedicated for a validation of the trained model. If the parameters is not set,a validation frame created via the ‘splitRatio’ parameter. The parameter is not serializable!
Scala default value:
null
; Python default value:None
- weightCol
Column with observation weights. Giving some observation a weight of zero is equivalent to excluding it from the dataset; giving an observation a relative weight of 2 is equivalent to repeating that row twice. Negative weights are not allowed. Note: Weights are per-row observation weights and do not increase the size of the data frame. This is typically the number of times a row is repeated, but non-integer values are supported as well. During training, rows with higher weights matter more, due to the larger loss function pre-factor. If you set weight = 0 for a row, the returned prediction frame at that row is zero and this is incorrect. To get an accurate prediction, remove all rows with weight == 0.
Scala default value:
null
; Python default value:None
Also available on the trained model.
- withContributions
Enables or disables generating a sub-column of detailedPredictionCol containing Shapley values of original features.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- withLeafNodeAssignments
Enables or disables computation of leaf node assignments.
Scala default value:
false
; Python default value:False
Also available on the trained model.
- withStageResults
Enables or disables computation of stage results.
Scala default value:
false
; Python default value:False
Also available on the trained model.