Simplifying logic in training_set file object and checking user given labels