def serving_input_fn()

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

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

def serving_input_fn():
  receiver_tensor = {'input': tf.placeholder(shape=[None, None], dtype=tf.int64)}
  features = {'movie_ids': receiver_tensor['input']}
  return tf.estimator.export.ServingInputReceiver(features, receiver_tensor)