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.

Constructor Summary

Constructors

Constructor and Description
Trainer(smile.data.Attribute[] attributes, int maxNodes) Constructor.
Trainer(int maxNodes) Constructor.
Trainer(int numFeatures, int maxNodes) Constructor.

Method Summary

Modifier and Type	Method and 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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Trainer

public Trainer(int maxNodes)

Constructor.

Parameters:

maxNodes - the maximum number of leaf nodes in the tree.

Trainer

public Trainer(smile.data.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 class RegressionTrainer<double[]>

Parameters:

x - the training instances.

y - the training response values.

Returns:

a trained regression model.

train

public RegressionTree train(int[][] x,
                            double[] y)