restore tensorflow model and use it for prediction

Question

https://colab.research.google.com/github/tensorflow/adanet/blob/master/adanet/examples/tutorials/adanet_objective.ipynb#scrollTo=s5RuM582oxo_

I am modifying the code