diff --git a/README.md b/README.md index d5d81ee..473d6df 100644 --- a/README.md +++ b/README.md @@ -34,7 +34,7 @@ model.fit([x_train, y_train], batch_size=batch_size, epochs=epochs, verbose=1, - validation_data=(x_test, y_test), + validation_data=([x_test, y_test], y_test), callbacks=[ModelCheckpoint('model.hdf5', verbose=1, save_best_only=True)]) ```