Package smile.regression
Class RegressionTree.Trainer
- java.lang.Object
-
- smile.regression.RegressionTrainer<double[]>
-
- smile.regression.RegressionTree.Trainer
-
- Enclosing class:
- RegressionTree
public static class RegressionTree.Trainer extends RegressionTrainer<double[]>
Trainer for regression tree.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description RegressionTree.Trainer
setMaxNodes(int maxNodes)
Sets the maximum number of leaf nodes in the tree.RegressionTree.Trainer
setNodeSize(int nodeSize)
Sets the minimum size of leaf nodes.RegressionTree
train(double[][] x, double[] y)
Learns a regression model with given training data.RegressionTree
train(int[][] x, double[] y)
-
Methods inherited from class smile.regression.RegressionTrainer
setAttributes
-
-
-
-
Constructor Detail
-
Trainer
public Trainer(int maxNodes)
Constructor.- Parameters:
maxNodes
- the maximum number of leaf nodes in the tree.
-
Trainer
public Trainer(Attribute[] attributes, int maxNodes)
Constructor.- Parameters:
attributes
- the attributes of independent variable.maxNodes
- the maximum number of leaf nodes in the tree.
-
Trainer
public Trainer(int numFeatures, int maxNodes)
Constructor.- Parameters:
numFeatures
- the number of features.maxNodes
- the maximum number of leaf nodes in the tree.
-
-
Method Detail
-
setMaxNodes
public RegressionTree.Trainer setMaxNodes(int maxNodes)
Sets the maximum number of leaf nodes in the tree.- Parameters:
maxNodes
- the maximum number of leaf nodes in the tree.
-
setNodeSize
public RegressionTree.Trainer setNodeSize(int nodeSize)
Sets the minimum size of leaf nodes.- Parameters:
nodeSize
- the minimum size of leaf nodes..
-
train
public RegressionTree train(double[][] x, double[] y)
Description copied from class:RegressionTrainer
Learns a regression model with given training data.- Specified by:
train
in classRegressionTrainer<double[]>
- Parameters:
x
- the training instances.y
- the training response values.- Returns:
- a trained regression model.
-
train
public RegressionTree train(int[][] x, double[] y)
-
-