image_output_1.png
"""his = model.fit_generator(train_gen, 
                              epochs=10, 
                              steps_per_epoch=len(X_train)/BATCH_SIZE, 
                              validation_data=test_gen, 
                              validation_steps=len(X_test)/BATCH_SIZE)"""