diff --git a/image_recognition_keras_tutorial.py b/image_recognition_keras_tutorial.py index c89fafd..6c42a67 100644 --- a/image_recognition_keras_tutorial.py +++ b/image_recognition_keras_tutorial.py @@ -76,3 +76,11 @@ scores = model.evaluate(X_test, y_test, verbose=0) print("Accuracy: %.2f%%" % (scores[1]*100)) + +# Save a model +model.save('SavedModel/my_model.tflearn') +print("MODEL SAVED") + +# Load a model +model.load('SavedModel/my_model.tflearn') +print("MODEL LOADED")