public class GDBBinaryClassifierOnTreesTrainer extends GDBBinaryClassifierTrainer
GDBTrainer.GDBModel
DatasetTrainer.EmptyDatasetException
checkConvergenceStgyFactory, loss
envBuilder, environment
Constructor and Description |
---|
GDBBinaryClassifierOnTreesTrainer(double gradStepSize,
Integer cntOfIterations,
int maxDepth,
double minImpurityDecrease)
Constructs instance of GDBBinaryClassifierOnTreesTrainer.
|
Modifier and Type | Method and Description |
---|---|
protected @NotNull DecisionTreeRegressionTrainer |
buildBaseModelTrainer()
Returns regressor model trainer for one step of GDB.
|
protected GDBLearningStrategy |
getLearningStrategy()
Returns learning strategy.
|
int |
getMaxDepth()
Get the max depth.
|
double |
getMinImpurityDecrease()
Get the min impurity decrease.
|
boolean |
isUsingIdx()
Get the using index structure property instead of using sorting during the learning process.
|
GDBBinaryClassifierOnTreesTrainer |
setMaxDepth(int maxDepth)
Set up the max depth.
|
GDBBinaryClassifierOnTreesTrainer |
setMinImpurityDecrease(double minImpurityDecrease)
Set up the min impurity decrease.
|
GDBBinaryClassifierOnTreesTrainer |
withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
Changes learning Environment.
|
GDBBinaryClassifierOnTreesTrainer |
withUsingIdx(boolean usingIdx)
Set useIndex parameter and returns trainer instance.
|
externalLabelToInternal, internalLabelToExternal, learnLabels
computeInitialValue, fitWithInitializedDeployingContext, isUpdateable, updateModel, withCheckConvergenceStgyFactory
fit, fit, fit, fit, fit, fit, getLastTrainedModelOrThrowEmptyDatasetException, identityTrainer, learningEnvironment, update, update, update, update, update, withConvertedLabels
public GDBBinaryClassifierOnTreesTrainer(double gradStepSize, Integer cntOfIterations, int maxDepth, double minImpurityDecrease)
gradStepSize
- Gradient step size.cntOfIterations
- Count of iterations.maxDepth
- Max depth.minImpurityDecrease
- Min impurity decrease.@NotNull protected @NotNull DecisionTreeRegressionTrainer buildBaseModelTrainer()
buildBaseModelTrainer
in class GDBTrainer
protected GDBLearningStrategy getLearningStrategy()
getLearningStrategy
in class GDBTrainer
public GDBBinaryClassifierOnTreesTrainer withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
withEnvironmentBuilder
in class GDBBinaryClassifierTrainer
envBuilder
- Learning environment builder.public GDBBinaryClassifierOnTreesTrainer withUsingIdx(boolean usingIdx)
usingIdx
- Use index.public int getMaxDepth()
public GDBBinaryClassifierOnTreesTrainer setMaxDepth(int maxDepth)
maxDepth
- The parameter value.public double getMinImpurityDecrease()
public GDBBinaryClassifierOnTreesTrainer setMinImpurityDecrease(double minImpurityDecrease)
minImpurityDecrease
- The parameter value.public boolean isUsingIdx()
GridGain In-Memory Computing Platform : ver. 8.9.14 Release Date : November 5 2024