Everything looks fine, but I looked in your repository, I think there should be labels for each train, val, test data separately.