y_pred = model.predict(test_images)
y_pred = y_pred.argmax(axis = 1)
cm = confusion_matrix(test_labels, y_pred)y_pred의 각행에서(axis = 1) 가장 큰 값의 인덱스를 argmax로 구하면 레이블 인코딩이 된다.
'인공지능 > 텐서플로우' 카테고리의 다른 글
| 텐서플로우의 콜백클래스를 이용해서, 원하는 조건이 되면 학습을 멈추게 하기 (0) | 2022.06.13 |
|---|---|
| 에포크, 학습데이터/밸리데이션데이터와 오버피팅 (0) | 2022.06.13 |
| 분류의 문제에서 loss 셋팅하는 방법 (0) | 2022.06.13 |
| 여러 클래스로 분류할 때, 아웃풋 노드에서 사용하는 액티베이션 함수 softmax (0) | 2022.06.13 |
| Flatten 라이브러리 사용하는 이유 (0) | 2022.06.13 |