saving trained model.

AI-MLProjects · May 20, 2021 · 11e8d5d · 11e8d5d
1 parent fe32cdb
commit 11e8d5d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/RockPaperScissors.py b/RockPaperScissors.py
@@ -269,4 +269,6 @@ def on_epoch_end(self, epoch, logs={}):
     validation_steps=validation_steps,
     verbose=1,
     callbacks=[callbacks]
-)
+)
+
+model.save("rock-paper-scissors-trained.h5")