Class StochasticForests.TreeRegression

    • Field Detail

      • counter

        private int[] counter
      • sums

        private double[] sums
    • Constructor Detail

      • TreeRegression

        public TreeRegression()
      • TreeRegression

        public TreeRegression​(java.util.Vector<java.util.Vector<java.lang.Integer>> child_nodeIDs,
                              java.util.Vector<java.lang.Integer> split_varIDs,
                              java.util.Vector<java.lang.Double> split_values)
    • Method Detail

      • getPredictionTerminalNodeID

        public int getPredictionTerminalNodeID​(int sampleID)
      • getPrediction

        public double getPrediction​(int sampleID)
      • estimate

        public double estimate​(int nodeID)
      • splitNodeInternal

        public boolean splitNodeInternal​(int nodeID,
                                         java.util.Vector<java.lang.Integer> possible_split_varIDs)
        Specified by:
        splitNodeInternal in class StochasticForests.Tree
      • findBestSplit

        public boolean findBestSplit​(int nodeID,
                                     java.util.Vector<java.lang.Integer> possible_split_varIDs)
      • findBestSplitValueSmallQ

        public void findBestSplitValueSmallQ​(int nodeID,
                                             int varID,
                                             double sum_node,
                                             int num_samples_node,
                                             double[] best_value,
                                             int[] best_varID,
                                             double[] best_decrease)
      • findBestSplitValueLargeQ

        public void findBestSplitValueLargeQ​(int nodeID,
                                             int varID,
                                             double sum_node,
                                             int num_samples_node,
                                             double[] best_value,
                                             int[] best_varID,
                                             double[] best_decrease)
      • findBestSplitValueUnordered

        public void findBestSplitValueUnordered​(int nodeID,
                                                int varID,
                                                double sum_node,
                                                int num_samples_node,
                                                double[] best_value,
                                                int[] best_varID,
                                                double[] best_decrease)
      • findBestSplitMaxstat

        public boolean findBestSplitMaxstat​(int nodeID,
                                            java.util.Vector<java.lang.Integer> possible_split_varIDs)
      • findBestSplitExtraTrees

        public boolean findBestSplitExtraTrees​(int nodeID,
                                               java.util.Vector<java.lang.Integer> possible_split_varIDs)
      • findBestSplitValueExtraTrees

        public void findBestSplitValueExtraTrees​(int nodeID,
                                                 int varID,
                                                 double sum_node,
                                                 int num_samples_node,
                                                 double[] best_value,
                                                 int[] best_varID,
                                                 double[] best_decrease)
      • findBestSplitValueExtraTreesUnordered

        public void findBestSplitValueExtraTreesUnordered​(int nodeID,
                                                          int varID,
                                                          double sum_node,
                                                          int num_samples_node,
                                                          double[] best_value,
                                                          int[] best_varID,
                                                          double[] best_decrease)
      • addImpurityImportance

        public void addImpurityImportance​(int nodeID,
                                          int varID,
                                          double decrease)