from tensorflow.contrib.learn.python.learn.estimators import model_fn as model_fn_lib

          GMM.SCORES: _streaming_sum(loss),
      }
      return model_fn_lib.ModelFnOps(mode=mode, predictions=predictions,
                                     eval_metric_ops=eval_metric_ops,
