왕현성
코딩발자취
왕현성
전체 방문자
오늘
어제
  • 코딩 (277)
    • Python (71)
    • Java (16)
    • MySQL (34)
    • 인공지능 (48)
      • 머신러닝 (16)
      • 딥러닝 (32)
    • 영상처리 (4)
    • Rest API (21)
    • Android Studio (25)
    • streamlit (13)
    • DevOps (22)
      • AWS (9)
      • PuTTY (5)
      • Git (4)
      • Serverless (2)
      • Docker (2)
    • IT 기술 용어 (6)
    • 디버깅 ( 오류 해결 과정 ) (17)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

인기 글

태그

  • matplotlib
  • alibidetect
  • get_long_description
  • pytorch
  • 영상처리역사
  • PYTHON
  • 딥러닝
  • maskimage
  • 영상처리
  • 비지도학습
  • OpenCV
  • numpy
  • alibi-detection
  • ckpt_file
  • encoding='utf-8'
  • 영상기술
  • 의료이미징
  • 컴퓨터비전
  • PIL
  • labelme
  • imageprocessing
  • tune()
  • TensorFlow
  • ComputerVision
  • pip install labelme
  • yolov8
  • labelme UnocodeDecodeError
  • UnboundLocalError
  • unsupervised
  • 기상탐사

최근 댓글

최근 글

티스토리

250x250
hELLO · Designed By 정상우.
왕현성
인공지능/딥러닝

딥러닝 : Tensorflow의 콜백클래스를 이용해서 원하는 조건이 되면 학습을 멈추게 하기

딥러닝 : Tensorflow의 콜백클래스를 이용해서 원하는 조건이 되면 학습을 멈추게 하기
인공지능/딥러닝

딥러닝 : Tensorflow의 콜백클래스를 이용해서 원하는 조건이 되면 학습을 멈추게 하기

2022. 12. 29. 11:20
728x90

https://hyunsungstory.tistory.com/161

 

딥러닝 : Fashion mnist 10개로 분류된 패션 이미지를 tensorflow를 이용해 분류하기 / Flatten , softmax , 분

1. 이미지와 행렬 모든 이미지 사진은 픽셀당 숫자로 되어있습니다. 0~255까지 되어있고 0이 검정색, 255가 흰색입니다. 그 숫자의 데이터 타입은 Uint8(Unsigned int) 이라고 적습니다. 먼저 검정부터 회

hyunsungstory.tistory.com

이전 게시글의 데이터셋을 이용합니다.

 

epochs가 무조건 많다고 좋은 것이 아님을 이 바로 전 게시글에서 확인할 수 있었습니다. ( 오버 피팅 )

 

그래서 이제 val_accuracy가 88%가 넘으면 멈추도록 하고싶다.

 

class myCallback(tf.keras.callbacks.Callback) :
  def on_epoch_end(self,epoch,logs={}) :
    if logs['val_accuracy'] > 0.88:
      print('\n내가 정한 정확도에 도달했으니, 학습을 멈춘다')
      self.model.stop_training = True

위와 같이 원하는 조건이 되면 학습을 멈추게하는 코드를 작성해주고

my_cb=myCallback()

이를 변수로 저장합니다.

 

def build_model():
  model = Sequential()
  model.add( Flatten()  )
  model.add( Dense(128, 'relu') )
  model.add( Dense(64, 'relu') )
  model.add( Dense(10, 'softmax'))
  model.compile('adam', 'sparse_categorical_crossentropy', ['accuracy'])
  return model
model = build_model()
epoch_history = model.fit(X_train,y_train,epochs=30,validation_split=0.2,callbacks=[my_cb])

모델링 이후 변수 저장후 학습을 진행하게 되면

epochs를 30으로 설정 했어도 val_accuracy가 88%가 넘는 6번 째 epochs에서 학습이 끝난 것을 확인할 수 있다.

'인공지능 > 딥러닝' 카테고리의 다른 글

딥러닝 : Tensorflow의 모델을 저장하고 불러오는 방법  (0) 2022.12.29
딥러닝 : Flatten()라이브러리 없이 이미지를 평탄화 하는 방법과 Validation_data= 파라미터 사용법  (0) 2022.12.29
딥러닝 : epochs의 횟수를 늘렸을 때 학습데이터/밸리데이션 데이터와 OverFitting  (0) 2022.12.29
딥러닝 : softmax로 나온 결과를 레이블 인코딩으로 바꾸는 방법  (0) 2022.12.29
딥러닝 : Fashion mnist 10개로 분류된 패션 이미지를 tensorflow를 이용해 분류하기 / Flatten , softmax , 분류의 문제 loss 셋팅법  (0) 2022.12.28
    '인공지능/딥러닝' 카테고리의 다른 글
    • 딥러닝 : Tensorflow의 모델을 저장하고 불러오는 방법
    • 딥러닝 : Flatten()라이브러리 없이 이미지를 평탄화 하는 방법과 Validation_data= 파라미터 사용법
    • 딥러닝 : epochs의 횟수를 늘렸을 때 학습데이터/밸리데이션 데이터와 OverFitting
    • 딥러닝 : softmax로 나온 결과를 레이블 인코딩으로 바꾸는 방법
    왕현성
    왕현성
    AI 머신비전 학습일지

    티스토리툴바

    단축키

    내 블로그

    내 블로그 - 관리자 홈 전환
    Q
    Q
    새 글 쓰기
    W
    W

    블로그 게시글

    글 수정 (권한 있는 경우)
    E
    E
    댓글 영역으로 이동
    C
    C

    모든 영역

    이 페이지의 URL 복사
    S
    S
    맨 위로 이동
    T
    T
    티스토리 홈 이동
    H
    H
    단축키 안내
    Shift + /
    ⇧ + /

    * 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.