Prediction Mode

Chapter Goals:

• Set up the regression function’s prediction code

A. Prediction values

For the prediction mode in the regression function, we initialize and return an EstimatorSpec object containing a dictionary with the model’s predictions. The model’s predictions need to be in 2-D tensor format, with ...