model.compile('adam', loss = 'categorical_crossentropy', metrics = ['accuracy'])
2개 분류의 문제일때는, 로스 펑션을 binary_crossentropy
3개 이상의 분류의 문제일때는? 2가지 경우가 있다!
첫번째 경우, y의 값이 레이블 인코딩으로 되어있는 경우 : sparse_categorical_crossentropy
두번째 경우, y의 값이 원핫 인코딩으로 되어있는 경우 : categorical_crossentropy
'인공지능 > 텐서플로우' 카테고리의 다른 글
이미지가 주어졌을 때, 컨볼루션한 결과로 나오는, feature map의 사이즈를 구하는 공식 (0) | 2022.06.14 |
---|---|
CNN의 convolution과 pooling (0) | 2022.06.14 |
레이블링된 y값을 to_categorical 함수 이용해서 원핫인코딩으로 바꾸기 (0) | 2022.06.14 |
텐서플로우의 모델 중, 네트워크와 웨이트를 따로 저장하고 불러오는 방법 (0) | 2022.06.14 |
텐서플로우의 모델을 저장하고 불러오는 방법 (0) | 2022.06.14 |