Balance Data

We need to define a strategy to balance data when creating the training data
This will be done, if required, on training

This work should be performed on the dataset pipeline during training phase.