def get_eval_spec()

in recommended-item-search/softmax_main.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def get_eval_spec():
  """Get eval spec for Estimator."""
  exporter = tf.estimator.LatestExporter(
      name=FLAGS.export_dir, exports_to_keep=FLAGS.keep_checkpoint_max,
      serving_input_receiver_fn=softmax_model.serving_input_fn)
  eval_input_fn = input_pipeline.generate_input_fn(
      file_pattern=FLAGS.eval_filename, batch_size=FLAGS.eval_batch_size,
      mode=tf.estimator.ModeKeys.EVAL)
  eval_spec = tf.estimator.EvalSpec(
      input_fn=eval_input_fn, steps=FLAGS.eval_steps,
      throttle_secs=FLAGS.eval_throttle_secs, exporters=exporter)
  return eval_spec