fix weird use of super

vecto-ai · Jan 22, 2019 · d13c2f7 · d13c2f7
1 parent 85200ac
commit d13c2f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vecto/benchmarks/categorization/categorization.py b/vecto/benchmarks/categorization/categorization.py
@@ -156,7 +156,7 @@ def collect_stats(self, data, vectors, labels):
                 word_entry = '{}. {}'.format(word_counter, word)
                 word_counter += 1
                 word_stats[word_entry] = self.process_stats(word_vector, centroid, category, predicted_category)
-        metric_scores = super(KMeansCategorization, self).compute_metics(predicted_labels, true_labels)
+        metric_scores = self.compute_metics(predicted_labels, true_labels)
         global_stats = self.process_global_stats(inertia, params, metric_scores, categories, predicted_labels,
                                                  true_labels)
         return dict(word_stats), global_stats