Fix bug in classifier summary print out

kboone · May 27, 2019 · d66aa49 · d66aa49
1 parent 430ec99
commit d66aa49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/avocado/classifier.py b/avocado/classifier.py
@@ -441,7 +441,7 @@ def train(self, dataset, num_folds=None, random_state=None, **kwargs):
             original_logloss = weighted_multi_logloss(
                 object_classes[original_mask],
                 predictions[original_mask],
-                object_weights=object_weights,
+                object_weights=object_weights[original_mask],
                 class_weights=self.class_weights
             )
             unweighted_original_logloss = weighted_multi_logloss(