Skip to content

21_22-J 38
21_22-J 38
Commit 6f13bc17 authored by W.D.R.P. Sandeepa's avatar W.D.R.P. Sandeepa
Browse files
Options

changes

parent a2a45bbd
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
backend/IT18218640/train.py
View file @ 6f13bc17
...@@ -112,11 +112,11 @@ def main(): ...@@ -112,11 +112,11 @@ def main():
history = model.fit(X_train, y_train, epochs=EPOCHS, batch_size=BATCH_SIZE, validation_data=(X_validation, y_validation), callbacks=[checkpoint]) history = model.fit(X_train, y_train, epochs=EPOCHS, batch_size=BATCH_SIZE, validation_data=(X_validation, y_validation), callbacks=[checkpoint])
# evaluate the model # evaluate the model
test_error, test_accuracy = model.evaluate(X_test, y_test) # test_error, test_accuracy = model.evaluate(X_test, y_test)
print(f"Test error: {test_error}, test accuracy: {test_accuracy}") # print(f"Test error: {test_error}, test accuracy: {test_accuracy}")
# save the model # save the model
# model.save(SAVE_MODEL_PATH) model.save(SAVE_MODEL_PATH)
show_graph(history) show_graph(history)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment