Improving the performance of weak supervision searches using data augmentation