Autoencoder Based Data Augmentation Python Code For Prime