From 015b4fa1e1fbdb33db1f0678b86c8329c275573e Mon Sep 17 00:00:00 2001 From: Nicolasticot Date: Mon, 28 Oct 2019 23:52:28 +0100 Subject: [PATCH] corrected minor bug --- DeepDrug.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DeepDrug.py b/DeepDrug.py index db9688f..aa85e5e 100644 --- a/DeepDrug.py +++ b/DeepDrug.py @@ -239,7 +239,7 @@ tf.test.is_gpu_available() # In[ ]: -history_mild_2mp = mild_model.fit(X_train, Y_train, validation_data=(X_test, Y_test), epochs=30, batch_size=32) +history_mild_2mp = my_model.fit(X_train, Y_train, validation_data=(X_test, Y_test), epochs=30, batch_size=32) my_model.save('new_model_e30_b32_t1000.h5')