public class DecisionTreeClassificationTrainer extends DecisionTree<GiniImpurityMeasure>
DatasetTrainer.EmptyDatasetException
usingIdx
envBuilder, environment
Constructor and Description |
---|
DecisionTreeClassificationTrainer()
Constructs a new decision tree classifier with default impurity function compressor
and default maxDeep = 5 and minImpurityDecrease = 0.
|
DecisionTreeClassificationTrainer(int maxDeep,
double minImpurityDecrease)
Constructs a new decision tree classifier with default impurity function compressor.
|
DecisionTreeClassificationTrainer(int maxDeep,
double minImpurityDecrease,
StepFunctionCompressor<GiniImpurityMeasure> compressor)
Constructs a new instance of decision tree classifier.
|
Modifier and Type | Method and Description |
---|---|
protected ImpurityMeasureCalculator<GiniImpurityMeasure> |
getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
Returns impurity measure calculator.
|
double |
getMaxDeep() |
double |
getMinImpurityDecrease() |
DecisionTreeClassificationTrainer |
withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
Changes learning Environment.
|
DecisionTreeClassificationTrainer |
withMaxDeep(Double maxDeep)
Set up the max deep of decision tree.
|
DecisionTreeClassificationTrainer |
withMinImpurityDecrease(Double minImpurityDecrease)
Set up the min impurity decrease of decision tree.
|
DecisionTreeClassificationTrainer |
withUseIndex(boolean useIdx)
Sets useIndex parameter and returns trainer instance.
|
fit, fitWithInitializedDeployingContext, isUpdateable, printTree, updateModel
fit, fit, fit, fit, fit, fit, getLastTrainedModelOrThrowEmptyDatasetException, identityTrainer, learningEnvironment, update, update, update, update, update, withConvertedLabels
public DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease)
maxDeep
- Max tree deep.minImpurityDecrease
- Min impurity decrease.public DecisionTreeClassificationTrainer()
public DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease, StepFunctionCompressor<GiniImpurityMeasure> compressor)
maxDeep
- Max tree deep.minImpurityDecrease
- Min impurity decrease.public DecisionTreeClassificationTrainer withMaxDeep(Double maxDeep)
maxDeep
- The parameter value.public DecisionTreeClassificationTrainer withMinImpurityDecrease(Double minImpurityDecrease)
minImpurityDecrease
- The parameter value.public DecisionTreeClassificationTrainer withUseIndex(boolean useIdx)
useIdx
- Use index.public double getMaxDeep()
public double getMinImpurityDecrease()
protected ImpurityMeasureCalculator<GiniImpurityMeasure> getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
getImpurityMeasureCalculator
in class DecisionTree<GiniImpurityMeasure>
dataset
- Dataset.public DecisionTreeClassificationTrainer withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
withEnvironmentBuilder
in class DecisionTree<GiniImpurityMeasure>
envBuilder
- Learning environment builder.
GridGain In-Memory Computing Platform : ver. 8.9.14 Release Date : November 5 2024