소스 검색

corrected minor bug

Nicolasticot 5 년 전
부모
커밋
015b4fa1e1
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      DeepDrug.py

+ 1 - 1
DeepDrug.py 파일 보기

@@ -239,7 +239,7 @@ tf.test.is_gpu_available()
239 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 243
 my_model.save('new_model_e30_b32_t1000.h5')
244 244
 
245 245