Custom Estimator, Keras

Custom Estimator

- 모델을 만들때 wrapping을 해줘서 estimator 형태로 들어가게 해줘야함

- estimator 안에 model_fn=custom_model을 적용하면 됨

- output 설정은 estimatorspec에 들어가게 됨

- 아래는 윗 그림의 그린 파트에 해당하는 내용임 

 > loss, optimizer, metric을 정해줌 → 그리고 estimatorSpec에서 위에서 정한 값이 나오게 되지

  > 참고 : https://www.tensorflow.org/tutorials/estimator/keras_model_to_estimator

 

Keras

- 모델을 만들기 쉽고, 직관적으로 이해하기 쉬움

  > 분산 처리를 위해 keras를 estimator에 넣음 → model_to_estimator

  > keras에 dataset을 input 하기 위해 create_feature_keras_input이라는 함수를 만듦 → 그 이유는 keras는 data input을 함수형태로 받지 않기 때문 <-> estimator는 input을 함수형태로 받아야함

 

keras 자체로 실행

 

 

keras를 estimator에 집어넣고 실행→ 분산 처리하고 싶으면 estimator를 써야함