File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -92,7 +92,7 @@ private double impurity(int count) {
92
92
93
93
// Returns a new Splitter containing only data where indices are true for the given predicate.
94
94
private Splitter mask (IntPredicate predicate ) {
95
- int [] indices = IntStream .range (0 , matrix . length ).filter (predicate ).toArray ();
95
+ int [] indices = IntStream .range (0 , size () ).filter (predicate ).toArray ();
96
96
double [][] newMatrix = new double [indices .length ][];
97
97
boolean [] newLabels = new boolean [indices .length ];
98
98
for (int i = 0 ; i < indices .length ; i += 1 ) {
@@ -104,7 +104,7 @@ private Splitter mask(IntPredicate predicate) {
104
104
105
105
// Returns an array of all possible feature splits for each matrix variable.
106
106
private FeatureSplit [] featureSplits () {
107
- int m = matrix . length ;
107
+ int m = size () ;
108
108
int n = matrix [0 ].length ;
109
109
FeatureSplit [] result = new FeatureSplit [n ];
110
110
for (int j = 0 ; j < n ; j += 1 ) {
You can’t perform that action at this time.
0 commit comments