Skip to content

Instantly share code, notes, and snippets.

@a-agmon
Created June 29, 2020 20:59
Show Gist options
  • Save a-agmon/51f2bf2fad9f7132190a11260f51b3fe to your computer and use it in GitHub Desktop.
Save a-agmon/51f2bf2fad9f7132190a11260f51b3fe to your computer and use it in GitHub Desktop.
optimizer = optimizers.Adam(lr=1e-2)
autoencoder.compile(optimizer=optimizer,
loss='mean_squared_error',
metrics=['accuracy'])
checkpointer = ModelCheckpoint(filepath="model_bin.h5",
verbose=0,
save_best_only=True)
tensorboard = TensorBoard(log_dir='./logs',
histogram_freq=0,
write_graph=True,
write_images=True)
history = autoencoder.fit(X_train, X_train,
epochs=nb_epoch,
batch_size=batch_size,
shuffle=True,
validation_data=(X_test, X_test),
verbose=1,
callbacks=[checkpointer, tensorboard]).history
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment