Solved by ptrblck in post #2 
              
              
                Since you are randomly applying the transformations you won’t have a guarantee to apply it, but could estimate the likelihood of getting a transformed sample for each class. 
Based on your description you could e.g. apply the transformations separately to at least one sample per class and let the ot…