Skip to content

Commit a35ea7f

Browse files
committed
Use getter
1 parent 6751924 commit a35ea7f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/lib/rpf_purify.cpp

+5-5
Original file line numberDiff line numberDiff line change
@@ -773,7 +773,7 @@ void extend_trees(
773773
for (std::vector<std::set<int>>::reverse_iterator key = keys.rbegin(); key != keys.rend(); ++key)
774774
{
775775
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();
777777
// check if number of dims same as current max_interaction
778778
if (curr_dims.size() == curr_max)
779779
{
@@ -782,9 +782,9 @@ void extend_trees(
782782
for (int feature_dim = 1; feature_dim <= feature_size; ++feature_dim)
783783
{
784784
// 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)
786786
{
787-
std::set<int> tree_dims = curr_tree->split_dims;
787+
std::set<int> tree_dims = curr_tree->get_split_dims();
788788
tree_dims.erase(tree_dims.find(feature_dim)); // remove current feature dim from current tree
789789
// check if tree with dimensions exists, if not create
790790
std::shared_ptr<DecisionTree> tree = treeExists(tree_dims, curr_family);
@@ -901,7 +901,7 @@ void RandomPlantedForest::purify_3_new()
901901
// ------------- create new trees -------------
902902

903903
extend_trees(feature_size, value_size, lim_list, curr_family, variables, X, grids, individuals, values, values_old);
904-
904+
905905
// ------------- purify -------------
906906
// iterate backwards through tree family
907907
int tree_index_t = curr_family.size() - 1;
@@ -1238,4 +1238,4 @@ void RandomPlantedForest::purify_anova_and_marginalize()
12381238
{
12391239
purify_anova();
12401240
purify_no_extrapolation_existing_grid();
1241-
}
1241+
}

0 commit comments

Comments
 (0)