나의 개발일지
CNN 모델 연습 기록 02 본문
학습 후기
일단 학습시간이 아주 오래 걸리고 학습데이터는 계속 정확도가 올라가지만
테스트데이터는 확률지지부진이고 일단 9번까지 돌리고 접음

수정한 모델
model = Sequential([
layers.Rescaling(1./255, input_shape=(56, 56, 1)),
layers.Conv2D(16, (3, 3), activation='relu'),
layers.MaxPooling2D(),
layers.Conv2D(32, (3, 3), activation='relu'),
layers.MaxPooling2D(),
layers.Conv2D(64, (3, 3), activation='relu'),
layers.MaxPooling2D(),
layers.Dropout(0.2),
layers.Flatten(),
layers.Dense(11172)
])
일단 추출된 특성이 분류하는 양보다 적어도 문제는 없다고 해서 다시 낮췄다.
보통은 128 => 64 => 10 이런 식으로 낮추는 게 보편적인 방법인 거 같음
확실히 학습시간도 짧고 정확도도 비슷한 거 같아서 일단은 돌려보겠다.
'기타 > AI연습 기록' 카테고리의 다른 글
CNN 모델 연습 기록 04 (0) | 2024.01.19 |
---|---|
CNN 모델 연습 기록 03 (0) | 2024.01.17 |
CNN 모델 연습 기록 01 (0) | 2024.01.09 |