기타/AI연습 기록
CNN 모델 연습 기록 02
인공지능싱글톤
2024. 1. 10. 20:42
학습 후기
일단 학습시간이 아주 오래 걸리고 학습데이터는 계속 정확도가 올라가지만
테스트데이터는 확률지지부진이고 일단 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 이런 식으로 낮추는 게 보편적인 방법인 거 같음
확실히 학습시간도 짧고 정확도도 비슷한 거 같아서 일단은 돌려보겠다.