Set class for containment of training and test data