Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created November 24, 2020 08:35
Show Gist options
  • Save amankharwal/7dd7a81bff5a2340822849fb5d9272f3 to your computer and use it in GitHub Desktop.
Save amankharwal/7dd7a81bff5a2340822849fb5d9272f3 to your computer and use it in GitHub Desktop.
model.compile(optimizer=Adam(lr=0.0001),loss='categorical_crossentropy',metrics=['accuracy'])
batch_size=32
epochs=64
history = model.fit_generator(datagen.flow(X_train,y_train, batch_size=batch_size),
epochs = epochs,
validation_data = (X_test,y_test),
verbose = 1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment