Class RegressionTree.Trainer

    • 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 class RegressionTrainer<double[]>
        Parameters:
        x - the training instances.
        y - the training response values.
        Returns:
        a trained regression model.