0bf56c5e8db53e152e7095087499d072018283e6,src/test.py,,main,#Any#,159

Before Change

    // Initialize the classifier
    classifier = tf.estimator.Estimator(model_fn=model_fn, 
                                        model_dir=FLAGS.model,
                                        config=tf.estimator.RunConfig(
                                            session_config=
                                            _get_session_config()))

    predictions = list(classifier.predict(input_fn=lambda: _get_input_stream()))
    print("Predictions: {}" .format(str(predictions)))

After Change

    
def main(argv=None):

    custom_config = tf.estimator.RunConfig(session_config=_get_session_config())
    
   

    // Initialize the classifier
    classifier = tf.estimator.Estimator(model_fn=model_fn, 
                                        model_dir=FLAGS.model,
                                        config=custom_config)

    predictions = classifier.predict(input_fn=lambda: _get_input_stream())
    print(next(predictions))

Instances