@@ -773,7 +773,7 @@ void extend_trees(
773
773
for (std::vector<std::set<int >>::reverse_iterator key = keys.rbegin (); key != keys.rend (); ++key)
774
774
{
775
775
auto &curr_tree = curr_family[(*key)];
776
- std::set<int > curr_dims = curr_tree->split_dims ;
776
+ std::set<int > curr_dims = curr_tree->get_split_dims () ;
777
777
// check if number of dims same as current max_interaction
778
778
if (curr_dims.size () == curr_max)
779
779
{
@@ -782,9 +782,9 @@ void extend_trees(
782
782
for (int feature_dim = 1 ; feature_dim <= feature_size; ++feature_dim)
783
783
{
784
784
// continue only if dim in current tree
785
- if (curr_tree->split_dims .count (feature_dim) != 0 )
785
+ if (curr_tree->get_split_dims () .count (feature_dim) != 0 )
786
786
{
787
- std::set<int > tree_dims = curr_tree->split_dims ;
787
+ std::set<int > tree_dims = curr_tree->get_split_dims () ;
788
788
tree_dims.erase (tree_dims.find (feature_dim)); // remove current feature dim from current tree
789
789
// check if tree with dimensions exists, if not create
790
790
std::shared_ptr<DecisionTree> tree = treeExists (tree_dims, curr_family);
@@ -901,7 +901,7 @@ void RandomPlantedForest::purify_3_new()
901
901
// ------------- create new trees -------------
902
902
903
903
extend_trees (feature_size, value_size, lim_list, curr_family, variables, X, grids, individuals, values, values_old);
904
-
904
+
905
905
// ------------- purify -------------
906
906
// iterate backwards through tree family
907
907
int tree_index_t = curr_family.size () - 1 ;
@@ -1238,4 +1238,4 @@ void RandomPlantedForest::purify_anova_and_marginalize()
1238
1238
{
1239
1239
purify_anova ();
1240
1240
purify_no_extrapolation_existing_grid ();
1241
- }
1241
+ }
0 commit comments