인공지능/텐서플로우

분류의 문제에서 loss 함수를 categorical_crossentropy로 설정할 상황

iminu 2022. 6. 14. 14:09
model.compile('adam', loss = 'categorical_crossentropy', metrics = ['accuracy'])

2개 분류의 문제일때는, 로스 펑션을 binary_crossentropy
3개 이상의 분류의 문제일때는? 2가지 경우가 있다!
첫번째 경우, y의 값이 레이블 인코딩으로 되어있는 경우 : sparse_categorical_crossentropy
두번째 경우, y의 값이 원핫 인코딩으로 되어있는 경우 : categorical_crossentropy