# 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