콜백 함수
유용한 콜백 함수 1 2 3 4 5 6 7 8 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]) 추가 cs 신경망이 한번 돌때 accuracy가 측정이 되는데, 내가 원하는 accuracy수치에 도달하였을 때 학습을 멈춰주는 함수이다. 1 2 3 4 5 6 epoch_history = model.fit(train_generator, ste..
coding/딥러닝
2021. 3. 2. 22:25