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