diff --git a/python/boomer/common/cpp/statistics.h b/python/boomer/common/cpp/statistics.h index fd86786b63..4078065427 100644 --- a/python/boomer/common/cpp/statistics.h +++ b/python/boomer/common/cpp/statistics.h @@ -269,7 +269,7 @@ class AbstractStatistics : virtual public IMatrix { * rule. * * @param statisticIndex The index of the statistic to be updated - * @param head A pointer to an object of type `Prediction`, representing the predictions of the + * @param prediction A pointer to an object of type `Prediction`, representing the predictions of the * newly induced rule */ virtual void applyPrediction(uint32 statisticIndex, Prediction* prediction) = 0; diff --git a/python/boomer/common/rule_induction.pyx b/python/boomer/common/rule_induction.pyx index 9da4dc7c77..fd7011e3e0 100644 --- a/python/boomer/common/rule_induction.pyx +++ b/python/boomer/common/rule_induction.pyx @@ -744,8 +744,9 @@ cdef inline void __recalculate_predictions(AbstractStatistics* statistics, uint3 for r in range(num_statistics): if covered_statistics_mask[r] == covered_statistics_target: statistics_subset_ptr.get().addToSubset(r, 1) - prediction = head_refinement.calculatePrediction(statistics_subset_ptr.get(), False, False) - updated_scores = prediction.predictedScores_ - for c in range(num_predictions): - predicted_scores[c] = updated_scores[c] + prediction = head_refinement.calculatePrediction(statistics_subset_ptr.get(), False, False) + updated_scores = prediction.predictedScores_ + + for c in range(num_predictions): + predicted_scores[c] = updated_scores[c]