chore: closes #21, #20

2023-09-27 21:20:39 +01:00
parent eab788aabd
commit 194277297f
9 changed files with 465 additions and 94 deletions
--- a/views/py/python_model_template.py
+++ b/views/py/python_model_template.py
@@ -46,7 +46,7 @@ model.compile(
    optimizer=tf.keras.optimizers.Adam(),
    metrics=['accuracy'])

-his = model.fit(dataset, validation_data= dataset_validation, epochs=50)
+his = model.fit(dataset, validation_data= dataset_validation, epochs=70)

 acc = his.history["accuracy"]

@@ -54,4 +54,6 @@ f = open("accuracy.val", "w")
 f.write(str(acc[-1]))
 f.close()

-model.save("model.keras")
+tf.saved_model.save(model, "model")
+
+# model.save("model.keras", save_format="tf")