Skip to content

Instantly share code, notes, and snippets.

@monogenea
Last active April 6, 2020 05:53
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save monogenea/db7ba8cd935e879a67df9d5f16e044ab to your computer and use it in GitHub Desktop.
Save monogenea/db7ba8cd935e879a67df9d5f16e044ab to your computer and use it in GitHub Desktop.
# Print summary
summary(model)
model %>% compile(optimizer = optimizer_adam(decay = 1e-5),
loss = "categorical_crossentropy",
metrics = "accuracy")
history <- fit(model, x = train$X, y = train$Y,
batch_size = 16, epochs = 50,
validation_data = list(val$X, val$Y))
plot(history)
# Save model
# model %>% save_model_hdf5("model.h5")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment