iter

scikit-learn · Sep 12, 2019 · 2f27d59 · 2f27d59
1 parent ddafd73
commit 2f27d59
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/sklearn/ensemble/weight_boosting.py b/sklearn/ensemble/weight_boosting.py
@@ -1035,8 +1035,10 @@ def _boost(self, iboost, X, y, sample_weight, random_state):
 
         error_vect = np.abs(y_predict - y)
         sample_mask = sample_weight > 0
-        error_max = error_vect[sample_mask].max()
+        masked_sample_weight = sample_weight[sample_mask]
+        masked_error_vector = error_vect[sample_mask]
 
+        error_max = masked_error_vector.max()
         if error_max != 0:
             error_vect /= error_max
 
@@ -1046,8 +1048,7 @@ def _boost(self, iboost, X, y, sample_weight, random_state):
             error_vect = 1. - np.exp(- error_vect)
 
         # Calculate the average loss
-        estimator_error = (sample_weight[sample_mask] *
-                           error_vect[sample_mask]).sum()
+        estimator_error = (masked_sample_weight * masked_error_vector).sum()
 
         if estimator_error <= 0:
             # Stop if fit is perfect
@@ -1066,8 +1067,8 @@ def _boost(self, iboost, X, y, sample_weight, random_state):
 
         if not iboost == self.n_estimators - 1:
             sample_weight[sample_mask] *= np.power(
-                beta,
-                (1. - error_vect[sample_mask]) * self.learning_rate)
+                beta, (1. - masked_error_vector) * self.learning_rate
+            )
 
         return sample_weight, estimator_weight, estimator_error