public class SharedTreeParametersV3 extends ModelParametersSchemaV3
| Modifier and Type | Field and Description |
|---|---|
boolean |
balanceClasses
Balance training data class counts via over/under-sampling (for imbalanced data).
|
boolean |
buildTreeOneNode
Run on one node only; no network overhead but fewer cpus used.
|
boolean |
calibrateModel
Use Platt Scaling (default) or Isotonic Regression to calculate calibrated class probabilities.
|
FrameKeyV3 |
calibrationFrame
Data for model calibration
|
TreeCalibrationHelperCalibrationMethod |
calibrationMethod
Calibration method to use
|
boolean |
checkConstantResponse
Check if response column is constant.
|
float[] |
classSamplingFactors
Desired over/under-sampling ratios per class (in lexicographic order).
|
double |
colSampleRateChangePerLevel
Relative change of the column sampling rate for every level (must be > 0.0 and <= 2.0)
|
double |
colSampleRatePerTree
Column sample rate per tree (from 0.0 to 1.0)
|
TreeSharedTreeModelSharedTreeParametersHistogramType |
histogramType
What type of histogram to use for finding optimal split points
|
java.lang.String |
inTrainingCheckpointsDir
Create checkpoints into defined directory while training process is still running.
|
int |
inTrainingCheckpointsTreeInterval
Checkpoint the model after every so many trees.
|
float |
maxAfterBalanceSize
Maximum relative size of the training data after balancing class counts (can be less than 1.0).
|
int |
maxConfusionMatrixSize
[Deprecated] Maximum size (# classes) for confusion matrices to be printed in the Logs
|
int |
maxDepth
Maximum tree depth (0 for unlimited).
|
double |
minRows
Fewest allowed (weighted) observations in a leaf.
|
double |
minSplitImprovement
Minimum relative improvement in squared error reduction for a split to happen
|
int |
nbins
For numerical columns (real/int), build a histogram of (at least) this many bins, then split at the best point
|
int |
nbinsCats
For categorical columns (factors), build a histogram of this many bins, then split at the best point.
|
int |
nbinsTopLevel
For numerical columns (real/int), build a histogram of (at most) this many bins at the root level, then decrease
by factor of two per level
|
int |
ntrees
Number of trees.
|
double |
r2Stopping
r2_stopping is no longer supported and will be ignored if set - please use stopping_rounds, stopping_metric and
stopping_tolerance instead.
|
double[] |
sampleRatePerClass
A list of row sample rates per class (relative fraction for each class, from 0.0 to 1.0), for each tree
|
int |
scoreTreeInterval
Score the model after every so many trees.
|
long |
seed
Seed for pseudo random number generator (if applicable)
|
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 |
|---|
SharedTreeParametersV3()
Public constructor
|
| Modifier and Type | Method and Description |
|---|---|
java.lang.String |
toString()
Return the contents of this object as a JSON String.
|
@SerializedName(value="balance_classes") public boolean balanceClasses
@SerializedName(value="class_sampling_factors") public float[] classSamplingFactors
@SerializedName(value="max_after_balance_size") public float maxAfterBalanceSize
@SerializedName(value="max_confusion_matrix_size") public int maxConfusionMatrixSize
public int ntrees
@SerializedName(value="max_depth") public int maxDepth
@SerializedName(value="min_rows") public double minRows
public int nbins
@SerializedName(value="nbins_top_level") public int nbinsTopLevel
@SerializedName(value="nbins_cats") public int nbinsCats
@SerializedName(value="r2_stopping") public double r2Stopping
public long seed
@SerializedName(value="build_tree_one_node") public boolean buildTreeOneNode
@SerializedName(value="sample_rate_per_class") public double[] sampleRatePerClass
@SerializedName(value="col_sample_rate_per_tree") public double colSampleRatePerTree
@SerializedName(value="col_sample_rate_change_per_level") public double colSampleRateChangePerLevel
@SerializedName(value="score_tree_interval") public int scoreTreeInterval
@SerializedName(value="min_split_improvement") public double minSplitImprovement
@SerializedName(value="histogram_type") public TreeSharedTreeModelSharedTreeParametersHistogramType histogramType
@SerializedName(value="calibrate_model") public boolean calibrateModel
@SerializedName(value="calibration_frame") public FrameKeyV3 calibrationFrame
@SerializedName(value="calibration_method") public TreeCalibrationHelperCalibrationMethod calibrationMethod
@SerializedName(value="check_constant_response") public boolean checkConstantResponse
@SerializedName(value="in_training_checkpoints_dir") public java.lang.String inTrainingCheckpointsDir
@SerializedName(value="in_training_checkpoints_tree_interval") public int inTrainingCheckpointsTreeInterval
public java.lang.String toString()
toString in class ModelParametersSchemaV3