diff --git a/main.py b/main.py index a0b6b2b772f7e447cfa7f9040dc5980917c0ca2f..aeb8c73999e87a13bb5d5d5935059f56ab9cab07 100644 --- a/main.py +++ b/main.py @@ -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):