Commit 69e6cb7b authored by Sintal's avatar Sintal
Browse files

Update main.py

parent 51b56bf0
...@@ -114,7 +114,7 @@ def train_and_save_model(model, X_train, y_train, X_test, y_test): ...@@ -114,7 +114,7 @@ def train_and_save_model(model, X_train, y_train, X_test, y_test):
print("Start Training...") print("Start Training...")
history = model.fit(X_train, y_train, epochs=35, validation_data=(X_test, y_test), callbacks=[tensorboard_callback]) history = model.fit(X_train, y_train, epochs=35, validation_data=(X_test, y_test), callbacks=[tensorboard_callback])
# Saving the trained model to avoid re-training # Saving the trained model to avoid re-training
model.save(constants.TRAINED_MODEL) #model.save(constants.TRAINED_MODEL)
return history return history
def model_predict(model, X_test, y_test): def model_predict(model, X_test, y_test):
......
Supports Markdown
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