nzimmermann
/
deepdrug3D
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

corrected minor bug

master
Nicolasticot 2019-10-28 23:52:28 +01:00
parent bb9a7f549e
commit 015b4fa1e1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
DeepDrug.py
View File

@ -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')