cleaning up after fit

Author: mathias-von-ottenbreit

cpp/APLRRegressor.h

@@ -422,6 +422,7 @@ void APLRRegressor::determine_interactions_to_consider()
                 if(!(terms[sorted_latest_split_point_errors_indices[i]]==interaction))
                 {
                     interaction.given_terms.push_back(terms[sorted_latest_split_point_errors_indices[i]]);
+                    interaction.given_terms[interaction.given_terms.size()-1].coefficient_steps.resize(0); //freeing memory
                     bool already_exists{false};
                     for (size_t k = 0; k < terms_eligible_current.size(); ++k)
                     {
@@ -836,6 +837,10 @@ void APLRRegressor::cleanup_after_fit()
     distributed_terms.clear();
     interactions_to_consider.clear();
     error_index_for_interactions_to_consider.resize(0);
+    for (size_t i = 0; i < terms.size(); ++i)
+    {
+        terms[i].clean_up_after_fit();
+    }
 }
 
 VectorXd APLRRegressor::predict(const MatrixXd &X)

cpp/term.h

@@ -54,6 +54,7 @@ class Term
     void calculate_coefficient_and_error_on_discretized_data(bool direction_right, double split_point);
     void estimate_coefficient_and_error_on_all_data();
     void clean_up_after_estimate_split_point();
+    void clean_up_after_fit();
 
 public:
     //fields
@@ -593,6 +594,16 @@ void Term::clean_up_after_estimate_split_point()
     errors_initial.resize(0);
 }
 
+void Term::clean_up_after_fit()
+{
+    bins_start_index.clear();
+    bins_end_index.clear();
+    bins_split_points_left.clear();
+    bins_split_points_right.clear();
+    values_discretized.resize(0);
+    sample_weight_discretized.resize(0);
+}
+
 VectorXd Term::calculate_prediction_contribution(const MatrixXd &X)
 {
     VectorXd values{calculate(X)};