Advertisement
Guest User

Untitled

a guest
Feb 26th, 2017
94
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.94 KB | None | 0 0
  1. model.compile(optimizer='rmsprop',
  2. loss={'main_output': 'mse', 'aux_output_gender':'categorical_crossentropy','aux_output_smile':'categorica_crossentropy','aux_output_glass':'categorical_crossentropy','aux_output_pose': 'categorical_crossentropy'},
  3. loss_weights={'main_output': 1.0, 'aux_output_gender': 0.1,'aux_output_smile': 0.1, 'aux_output_glass': 0.1,'aux_output_pose': 0.1})
  4.  
  5.  
  6. early_stopping1 =EarlyStopping(monitor='val_aux_output_pose_loss', patience=2)
  7.  
  8. history = model.fit({'main_input': X},
  9. {'main_output': y, 'aux_output_gender': y_aux_gender,'aux_output_smile': y_aux_smile,'aux_output_glass': y_aux_glass,'aux_output_pose': y_aux_pose}, validation_data=({'main_input': X_Test},{'main_output': y_test, 'aux_output_gender': y_aux_gender_test,'aux_output_smile': y_aux_smile_test,'aux_output_glass': y_aux_glass_test,'aux_output_pose': y_aux_pose_test}),
  10. nb_epoch=20, batch_size=16, callbacks=[early_stopping1])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement