Skip to content

Commit 2100c1b

Browse files
committed
Made variable names match
1 parent b9949a7 commit 2100c1b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/forest.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -772,7 +772,7 @@ cpp11::writable::integers get_overall_split_counts_active_forest_cpp(cpp11::exte
772772
for (int j = 0; j < split_nodes.size(); j++) {
773773
auto node_id = split_nodes.at(j);
774774
auto feature_split = tree->SplitIndex(node_id);
775-
output.at(split_feature)++;
775+
output.at(feature_split)++;
776776
}
777777
}
778778
return output;
@@ -789,7 +789,7 @@ cpp11::writable::integers get_granular_split_count_array_active_forest_cpp(cpp11
789789
for (int j = 0; j < split_nodes.size(); j++) {
790790
auto node_id = split_nodes.at(j);
791791
auto feature_split = tree->SplitIndex(node_id);
792-
output.at(split_feature*num_trees + i)++;
792+
output.at(feature_split*num_trees + i)++;
793793
}
794794
}
795795
return output;

0 commit comments

Comments
 (0)