Merge pull request #145 from ej0cl6/master

correct adaptive_active_learning bug
ntucllab · Jun 4, 2018 · e37e9ed · e37e9ed
2 parents aebd128 + ca94cec
commit e37e9ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libact/query_strategies/multilabel/adaptive_active_learning.py b/libact/query_strategies/multilabel/adaptive_active_learning.py
@@ -120,7 +120,7 @@ def make_query(self):
         candidate_idx_set = set()
         for b in self.betas:
             # score shape = (len(X_pool), )
-            score = separation_margin**b * label_cardinality**(1.-b)
+            score = uncertainty**b * label_cardinality**(1.-b)
             for idx in np.where(score == np.max(score))[0]:
                 candidate_idx_set.add(idx)