daa342d04c7fe19ff1a79775034c9c91e02d0e32,cloudml-template/template/trainer/model.py,,create_estimator,#Any#,117

Before Change

    print("creating a custom estimator")

    return tf.estimator.Estimator(model_fn=_model_fn,
                                  config=config)


// ***************************************************************************************
// THIS IS A HELPER FUNCTION TO CREATE GET THE HIDDEN LAYER UNITS

After Change

    print("creating a custom model...")

    estimator = tf.estimator.Estimator(model_fn=_model_fn, config=config)

    estimator = tf.contrib.estimator.add_metrics(estimator, metric_fn)

    return estimator


// ***************************************************************************************
// YOU NEED NOT TO CHANGE THESE HELPER FUNCTIONS USED FOR CONSTRUCTING THE MODELS

Instances