Class StochasticForests.TreeProbability

    • Field Detail

      • class_values

        java.util.Vector<java.lang.Double> class_values
      • response_classIDs

        java.util.Vector<java.lang.Integer> response_classIDs
      • sampleIDs_per_class

        java.util.Vector<java.util.Vector<java.lang.Integer>> sampleIDs_per_class
      • terminal_class_counts

        java.util.Vector<java.util.Vector<java.lang.Double>> terminal_class_counts
      • class_weights

        java.util.Vector<java.lang.Double> class_weights
      • counter

        int[] counter
      • counter_per_class

        int[] counter_per_class
    • Constructor Detail

      • TreeProbability

        public TreeProbability​(java.util.Vector<java.lang.Double> class_values,
                               java.util.Vector<java.lang.Integer> response_classIDs,
                               java.util.Vector<java.util.Vector<java.lang.Integer>> sampleIDs_per_class,
                               java.util.Vector<java.lang.Double> class_weights)
      • TreeProbability

        public TreeProbability​(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,
                               java.util.Vector<java.lang.Double> class_values,
                               java.util.Vector<java.lang.Integer> response_classIDs,
                               java.util.Vector<java.util.Vector<java.lang.Double>> terminal_class_counts)
    • Method Detail

      • getPrediction

        public java.util.Vector<java.lang.Double> getPrediction​(int sampleID)
      • getPredictionTerminalNodeID

        public int getPredictionTerminalNodeID​(int sampleID)
      • getTerminalClassCounts

        public java.util.Vector<java.util.Vector<java.lang.Double>> getTerminalClassCounts()
      • addToTerminalNodes

        public void addToTerminalNodes​(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,
                                             int num_classes,
                                             int[] class_counts,
                                             int num_samples_node,
                                             double[] best_value,
                                             int[] best_varID,
                                             double[] best_decrease)
      • findBestSplitValueLargeQ

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

        public void findBestSplitValueUnordered​(int nodeID,
                                                int varID,
                                                int num_classes,
                                                int[] class_counts,
                                                int num_samples_node,
                                                double[] best_value,
                                                int[] best_varID,
                                                double[] best_decrease)
      • findBestSplitExtraTrees

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

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

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

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