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

Update main.py

parent 51b56bf0
Pipeline #4396 failed with stage
in 59 seconds
......@@ -114,7 +114,7 @@ def train_and_save_model(model, X_train, y_train, X_test, y_test):
print("Start Training...")
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
model.save(constants.TRAINED_MODEL)
#model.save(constants.TRAINED_MODEL)
return history
def model_predict(model, X_test, y_test):
......
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