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 @@ -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;
You can’t perform that action at this time.
0 commit comments