콜백 함수

유용한 콜백 함수

class myCallback(tf.keras.callbacks.Callback) :
  def on_epoch_end(self, epoch, logs={}) :
    if(logs.get('accuracy') > 0.92) :
      print('\n정확도가 92% 이상이라 학습을 멈춥니다.')
      self.model.stop_training = True
my_cb = myCallback()
 
#학습 시 callbacks=[my_cb]) 추가

epoch_history = model.fit(train_generator, steps_per_epoch= 100,
                          epochs=15, 
                          verbose =1,
                          callbacks=[my_cb] #리스트 안에 넣어주면 실행가능
                          validation_data= validation_generator,
                          validation_steps = 50)