CNN 교육 조기 종료

Question

신호 분류에 CNN을 사용하고 있습니다 (두 클래스 : 환자 및 제어). 조기 정지 사용 유효성 검사 오류가 중지 될 때까지 교육을 중지하고 싶습니다.. 나는 에 이것을 구현하기 위해 파이썬 코드을 쓸 수 없다. 나는 이것을 시도했지만 더 이상 진행할 수 없었다.

validation = cross_entropy.eval(feed_dict= {x:valid_x,y_:valid_y,keep_prob:1}) 
validation = np.append(validation,cross_entropy.eval(feed_dict= {x:valid_x,y_:valid_y,keep_prob:1})) 

누구든지 코드를 도와 줄 수 있습니까?

Answer 1

당신은 keras.EarlyStopping를 사용할 수 있습니다

from keras.callbacks import EarlyStopping 
early_stopping = EarlyStopping(monitor='val_loss', patience=2) 
model.fit(x, y, validation_split=0.2, callbacks=[early_stopping]) 

이상적으로 val_loss 증가하지가 val_acc이 정체 될 때 때 훈련을 중지하는 것이 좋다. Kears는 val_acc가 향상 될 때 모델을 저장하기 때문에 필요할 때만 실행하고 중지하는 것이 좋습니다.