Machine-Learning-Assisted Antenna Optimization With Data Augmentation