Browse Source

corrected minor bug

Nicolasticot 4 years ago
parent
commit
015b4fa1e1
1 changed files with 1 additions and 1 deletions
  1. 1 1
      DeepDrug.py

+ 1 - 1
DeepDrug.py View File

239
 # In[ ]:
239
 # In[ ]:
240
 
240
 
241
 
241
 
242
-history_mild_2mp = mild_model.fit(X_train, Y_train, validation_data=(X_test, Y_test), epochs=30, batch_size=32)
242
+history_mild_2mp = my_model.fit(X_train, Y_train, validation_data=(X_test, Y_test), epochs=30, batch_size=32)
243
 my_model.save('new_model_e30_b32_t1000.h5')
243
 my_model.save('new_model_e30_b32_t1000.h5')
244
 
244
 
245
 
245