public class XGBoostParametersV3 extends ModelParametersSchemaV3
| Modifier and Type | Field and Description |
|---|---|
TreexgboostXGBoostModelXGBoostParametersBackend |
backend
Backend.
|
TreexgboostXGBoostModelXGBoostParametersBooster |
booster
Booster type
|
boolean |
buildTreeOneNode
Run on one node only; no network overhead but fewer cpus used.
|
boolean |
calibrateModel
Use Platt Scaling to calculate calibrated class probabilities.
|
FrameKeyV3 |
calibrationFrame
Calibration frame for Platt Scaling
|
double |
colsampleBylevel
(same as col_sample_rate) Column sample rate (from 0.0 to 1.0)
|
double |
colsampleBynode
Column sample rate per tree node (from 0.0 to 1.0)
|
double |
colsampleBytree
(same as col_sample_rate_per_tree) Column sample rate per tree (from 0.0 to 1.0)
|
double |
colSampleRate
(same as colsample_bylevel) Column sample rate (from 0.0 to 1.0)
|
double |
colSampleRatePerTree
(same as colsample_bytree) Column sample rate per tree (from 0.0 to 1.0)
|
TreexgboostXGBoostModelXGBoostParametersDMatrixType |
dmatrixType
Type of DMatrix.
|
double |
eta
(same as learn_rate) Learning rate (from 0.0 to 1.0)
|
float |
gamma
(same as min_split_improvement) Minimum relative improvement in squared error reduction for a split to happen
|
int[] |
gpuId
Which GPU(s) to use.
|
TreexgboostXGBoostModelXGBoostParametersGrowPolicy |
growPolicy
Grow policy - depthwise is standard GBM, lossguide is LightGBM
|
java.lang.String[][] |
interactionConstraints
A set of allowed column interactions.
|
double |
learnRate
(same as eta) Learning rate (from 0.0 to 1.0)
|
float |
maxAbsLeafnodePred
(same as max_delta_step) Maximum absolute value of a leaf node prediction
|
int |
maxBins
For tree_method=hist only: maximum number of bins
|
float |
maxDeltaStep
(same as max_abs_leafnode_pred) Maximum absolute value of a leaf node prediction
|
int |
maxDepth
Maximum tree depth (0 for unlimited).
|
int |
maxLeaves
For tree_method=hist only: maximum number of leaves
|
double |
minChildWeight
(same as min_rows) Fewest allowed (weighted) observations in a leaf.
|
double |
minRows
(same as min_child_weight) Fewest allowed (weighted) observations in a leaf.
|
float |
minSplitImprovement
(same as gamma) Minimum relative improvement in squared error reduction for a split to happen
|
KeyValueV3[] |
monotoneConstraints
A mapping representing monotonic constraints.
|
TreexgboostXGBoostModelXGBoostParametersDartNormalizeType |
normalizeType
For booster=dart only: normalize_type
|
int |
nthread
Number of parallel threads that can be used to run XGBoost.
|
int |
ntrees
(same as n_estimators) Number of trees.
|
boolean |
oneDrop
For booster=dart only: one_drop
|
boolean |
quietMode
Enable quiet mode
|
float |
rateDrop
For booster=dart only: rate_drop (0..1)
|
float |
regAlpha
L1 regularization
|
float |
regLambda
L2 regularization
|
double |
sampleRate
(same as subsample) Row sample rate per tree (from 0.0 to 1.0)
|
TreexgboostXGBoostModelXGBoostParametersDartSampleType |
sampleType
For booster=dart only: sample_type
|
java.lang.String |
saveMatrixDirectory
Directory where to save matrices passed to XGBoost library.
|
float |
scalePosWeight
Controls the effect of observations with positive labels in relation to the observations with negative labels on
gradient calculation.
|
int |
scoreTreeInterval
Score the model after every so many trees.
|
long |
seed
Seed for pseudo random number generator (if applicable)
|
float |
skipDrop
For booster=dart only: skip_drop (0..1)
|
double |
subsample
(same as sample_rate) Row sample rate per tree (from 0.0 to 1.0)
|
TreexgboostXGBoostModelXGBoostParametersTreeMethod |
treeMethod
Tree method
|
aucType, categoricalEncoding, checkpoint, customDistributionFunc, customMetricFunc, distribution, exportCheckpointsDir, foldAssignment, foldColumn, gainsliftBins, huberAlpha, ignoreConstCols, ignoredColumns, keepCrossValidationFoldAssignment, keepCrossValidationModels, keepCrossValidationPredictions, maxCategoricalLevels, maxRuntimeSecs, modelId, nfolds, offsetColumn, parallelizeCrossValidation, quantileAlpha, responseColumn, scoreEachIteration, stoppingMetric, stoppingRounds, stoppingTolerance, trainingFrame, tweediePower, validationFrame, weightsColumn| Constructor and Description |
|---|
XGBoostParametersV3()
Public constructor
|
| Modifier and Type | Method and Description |
|---|---|
java.lang.String |
toString()
Return the contents of this object as a JSON String.
|
public int ntrees
@SerializedName(value="max_depth") public int maxDepth
@SerializedName(value="min_rows") public double minRows
@SerializedName(value="min_child_weight") public double minChildWeight
@SerializedName(value="learn_rate") public double learnRate
public double eta
@SerializedName(value="sample_rate") public double sampleRate
public double subsample
@SerializedName(value="col_sample_rate") public double colSampleRate
@SerializedName(value="colsample_bylevel") public double colsampleBylevel
@SerializedName(value="col_sample_rate_per_tree") public double colSampleRatePerTree
@SerializedName(value="colsample_bytree") public double colsampleBytree
@SerializedName(value="colsample_bynode") public double colsampleBynode
@SerializedName(value="monotone_constraints") public KeyValueV3[] monotoneConstraints
@SerializedName(value="max_abs_leafnode_pred") public float maxAbsLeafnodePred
@SerializedName(value="max_delta_step") public float maxDeltaStep
@SerializedName(value="score_tree_interval") public int scoreTreeInterval
public long seed
@SerializedName(value="min_split_improvement") public float minSplitImprovement
public float gamma
public int nthread
@SerializedName(value="build_tree_one_node") public boolean buildTreeOneNode
@SerializedName(value="save_matrix_directory") public java.lang.String saveMatrixDirectory
@SerializedName(value="calibrate_model") public boolean calibrateModel
@SerializedName(value="calibration_frame") public FrameKeyV3 calibrationFrame
@SerializedName(value="max_bins") public int maxBins
@SerializedName(value="max_leaves") public int maxLeaves
@SerializedName(value="tree_method") public TreexgboostXGBoostModelXGBoostParametersTreeMethod treeMethod
@SerializedName(value="grow_policy") public TreexgboostXGBoostModelXGBoostParametersGrowPolicy growPolicy
public TreexgboostXGBoostModelXGBoostParametersBooster booster
@SerializedName(value="reg_lambda") public float regLambda
@SerializedName(value="reg_alpha") public float regAlpha
@SerializedName(value="quiet_mode") public boolean quietMode
@SerializedName(value="sample_type") public TreexgboostXGBoostModelXGBoostParametersDartSampleType sampleType
@SerializedName(value="normalize_type") public TreexgboostXGBoostModelXGBoostParametersDartNormalizeType normalizeType
@SerializedName(value="rate_drop") public float rateDrop
@SerializedName(value="one_drop") public boolean oneDrop
@SerializedName(value="skip_drop") public float skipDrop
@SerializedName(value="dmatrix_type") public TreexgboostXGBoostModelXGBoostParametersDMatrixType dmatrixType
public TreexgboostXGBoostModelXGBoostParametersBackend backend
@SerializedName(value="gpu_id") public int[] gpuId
@SerializedName(value="interaction_constraints") public java.lang.String[][] interactionConstraints
@SerializedName(value="scale_pos_weight") public float scalePosWeight
public java.lang.String toString()
toString in class ModelParametersSchemaV3