기타/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 이런 식으로 낮추는 게 보편적인 방법인 거 같음

확실히 학습시간도 짧고 정확도도 비슷한 거 같아서 일단은 돌려보겠다.