public class DecisionTreeRegressionTrainer extends DecisionTree<MSEImpurityMeasure>
DatasetTrainer.EmptyDatasetException
usingIdx
envBuilder, environment
Constructor and Description |
---|
DecisionTreeRegressionTrainer(int maxDeep,
double minImpurityDecrease)
Constructs a new decision tree regressor with default impurity function compressor.
|
DecisionTreeRegressionTrainer(int maxDeep,
double minImpurityDecrease,
StepFunctionCompressor<MSEImpurityMeasure> compressor)
Constructs a new decision tree regressor.
|
Modifier and Type | Method and Description |
---|---|
protected ImpurityMeasureCalculator<MSEImpurityMeasure> |
getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
Returns impurity measure calculator.
|
DecisionTreeRegressionTrainer |
withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
Changes learning Environment.
|
DecisionTreeRegressionTrainer |
withUsingIdx(boolean usingIdx)
Sets usingIdx 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 DecisionTreeRegressionTrainer(int maxDeep, double minImpurityDecrease)
maxDeep
- Max tree deep.minImpurityDecrease
- Min impurity decrease.public DecisionTreeRegressionTrainer(int maxDeep, double minImpurityDecrease, StepFunctionCompressor<MSEImpurityMeasure> compressor)
maxDeep
- Max tree deep.minImpurityDecrease
- Min impurity decrease.public DecisionTreeRegressionTrainer withUsingIdx(boolean usingIdx)
usingIdx
- Use index.protected ImpurityMeasureCalculator<MSEImpurityMeasure> getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
getImpurityMeasureCalculator
in class DecisionTree<MSEImpurityMeasure>
dataset
- Dataset.public DecisionTreeRegressionTrainer withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
withEnvironmentBuilder
in class DecisionTree<MSEImpurityMeasure>
envBuilder
- Learning environment builder.
GridGain In-Memory Computing Platform : ver. 8.9.14 Release Date : November 5 2024