Update _quantile_forest_fast.pyx

zillow · Sep 10, 2024 · 3e77b7f · 3e77b7f
1 parent b3e7731
commit 3e77b7f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/quantile_forest/_quantile_forest_fast.pyx b/quantile_forest/_quantile_forest_fast.pyx
@@ -667,9 +667,10 @@ cdef class QuantileForest:
                 leaf_weights = vector[double](n_train)
 
             for i in range(n_samples):
+                n_total_samples = 0
+                n_total_trees = 0
+
                 if weighted_leaves:
-                    n_total_samples = 0
-                    n_total_trees = 0
                     for j in range(n_trees):
                         if X_indices is None or X_indices[i, j] is True:
                             n_leaf_samples[j] = 0