dataloader.py line 67

[connorchen]

Hi,
labels_in_bag = labels_in_bag >= self.target_number
i think the purpose of the code is if the target exists in the labels_in_bag, so '>=' should be '=='
If my understanding is wrong, please correct me.
Thanks

[max-ilse]

Hi connorchen,
You are right. I just fixed that bug. Since we were using '9' as the target number that bug never caused issues. Thanks for pointing it out.
Max