How to train our custom dataset with certain labels in conditions, what should be the label and data format?