💾Save Model in HDF5

brianlechthaler · Sep 13, 2021 · 24de4a5 · 24de4a5
1 parent 5852289
commit 24de4a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -127,6 +127,6 @@ model.fit(
     callbacks=callbacks,
     validation_data=val_ds)
 
-# Save our model for later use.
-model.save(join('dataset',model_name))
+model.save(join('dataset',
+                f"{model_name}.h5"))
 ```
diff --git a/example.py b/example.py
@@ -112,4 +112,5 @@
     validation_data=val_ds)
 
 # Save our model for later use.
-model.save(join('dataset',model_name))
+model.save(join('dataset',
+                f"{model_name}.h5"))