fix bug: zero division

usc-isi-i2 · Aug 20, 2018 · 5bef246 · 5bef246
1 parent 36de463
commit 5bef246
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rltk/evaluation/ground_truth.py b/rltk/evaluation/ground_truth.py
@@ -241,7 +241,7 @@ def generate_stratified_negatives(self, dataset1: 'Dataset', dataset2: 'Dataset'
         for idx, s in enumerate(strata):
             stratum_id = str(idx)
             # nothing to pick
-            if s['p'] == 0 or s['n'] == 0:
+            if len(s['p']) == 0 or len(s['n']) == 0:
                 strata_weights[stratum_id] = 0.0
                 continue
             strata_weights[stratum_id] = float(len(s['p'])) / len(s['n'])