인프런 커뮤니티 질문&답변
작성한 질문수
학습한 모델을 이용하는 방법
작성
·
526
0
텐서플로우 시험 볼 때는 학습한 모델을 제출하고 끝나는데요.
실제로 학습시킨 모델을 이용해서 이미지를 분류해볼 수 있는 방법이 있을까요? (예. input으로 cat 이미지를 넣으면 'cat' 이라고 리턴해주는지)
Acc, Loss 값만으로 모델이 좋다/안좋다만 판별하니까 이게 어떤건지 잘 와닿지가 않아서요 ㅠ...
답변 1
0
런어데이
지식공유자
안녕하세요^^
학습된 모델로 실제로 테스트 해보기 위해서 다음과 같이 해주시면 됩니다
테스트 해볼 데이터셋(예를들면 강아지 혹은 고양이 이미지)을 학습하여 넣어준 전처리를 동일하게 진행하여 데이터셋 구성합니다(IDG 활용가능)
학습된 모델이 model 변수라면
model.predict(test_generator)로 예측할 수 있습니다.
감사합니다.