Issue #37 Save trained Keras model

keras-team · Aug 17, 2018 · a3ca627 · a3ca627
1 parent ccea64e
commit a3ca627
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/autokeras/image_classifier.py b/autokeras/image_classifier.py
@@ -287,4 +287,7 @@ def final_fit(self, x_train, y_train, x_test, y_test, trainer_args=None, retrain
 
     def get_best_model_id(self):
         """ Return an integer indicating the id of the best model."""
-        return self.load_searcher().get_best_model_id()
+        return self.load_searcher().get_best_model_id()
+
+    def export_keras_model(self, model_file_name):
+        self.load_searcher().load_best_model().produce_keras_model().save(model_file_name)