Tensorflow Estimator : 별도의 스크립트에서 predict() 함수 사용

Question

train/eval 정확도가 83-85 %에 도달 한 tf.Estimator를 사용하여 모델을 교육하고 평가했습니다. 이제 Estimator 클래스의 predict() 함수 호출을 사용하여 별도의 데이터 세트에서 내 모델을 테스트하고 싶습니다. 가급적 나는 별도의 스크립트로 이것을하고 싶습니다.

나는 SavedModel로 내보낼 필요가 있다고 말하는 this에 있었지만 실제로 필요한가? Estimator 클래스의 documentation을 보면 model_dir 매개 변수를 통해 내 검사 점 및 그래프 파일 경로를 전달할 수있는 것 같습니다. 이 경험이있는 사람이 있습니까? 유효성 검사를 위해 사용한 동일한 데이터 세트에서 모델을 실행할 때 유효성 검사 단계에서와 동일한 성능을 얻지 못합니다. :-(

Answer 1

model_fn 정의가 포함 된 별도의 파일이 필요하다고 생각합니다. 동일한 model_fn 정의와 같은 model_dir를 사용하여, 다른 스크립트에서 같은 추정 클래스를 인스턴스화합니다.

당신이 훈련을 계속 할 수 있도록 견적 API는 자체로 tf.Graph 정의와 최신 model.ckpt 파일을 복구하기 때문에 작동

, 평가 그리고 예측.