카테고리 없음
classification model, Test 할 때 predict 분류
교 향
2022. 11. 8. 18:15
출처 :
[Deep Learning][딥러닝] 딥러닝 구현
MNIST 이미지 분류 MNIST(Modified National Institute of Standards and Technology) database 흑백 손글씨 숫자 0-9까지 10개의 범주로 구분해놓은 데이터셋 하나의 이미지는 28 * 28 pixel 의 크기 6만개의 Train 이미지와 1
ysyblog.tistory.com
https://ysyblog.tistory.com/99
분류
- predict()
- 각 클래스 별 확률 반환
predict_class()- 클래스(범주값) 반환
- tensorflow 2.3 부터 deprecated 됨
- 이진 분류(binary classification)
- numpy.argmax(model.predict(x) > 0.5).astype("int32")
- 다중클래스 분류(multi-class classification)
- numpy.argmax(model.predict(x), axis=-1)