Train the malaria model

train_network.py

sess = K.get_session()
init = tf.global_variables_initializer()
sess.run(init)
STEP_SIZE_TRAIN = training_generator.samples // training_generator.batch_size
STEP_SIZE_VALID = validatation_generator.n // validatation_generator.batch_size
training_stats = CollectStats()
history = malaria_model.fit_generator(training_generator,
                                      epochs=50,
                                      steps_per_epoch=STEP_SIZE_TRAIN,
                                      validation_data=validatation_generator,
                                      validation_steps=STEP_SIZE_VALID,
                                      callbacks=[training_stats, early_stopping,
                                                 checkpoint, reduce_lr])