Breaking Change with TF 2.0

[santhh]

We currently use model = tf.contrib.estimator.forward_features(model,KEY_COLUMN) for the model creation which not supported for TF 2.0.
Need to upgrade the code for TF 2.0
https://www.tensorflow.org/guide/upgrade
@GoogleCloudPlatform/streaming-analytics-ai

[dhodun]

I wrote up a blog post and example notebook on how to do this in TF 2.x:

https://towardsdatascience.com/how-to-extend-a-keras-model-5effc083265c
https://github.com/GoogleCloudPlatform/training-data-analyst/blob/master/blogs/batch_predictions/batch_predictions_keras.ipynb

[santhh]

thanks @dhodun I will look. If you have some bandwidth-please feel free to contribute too.