b198dc46455cc467759b9d8b51c37809ec6248e0,boston/keras/trainer/model.py,,input_fn,#Any#Any#Any#Any#,59

Before Change

    A tf.estimator.
  
  // Default settings for training.
  num_epochs = None
  shuffle = True

  // Override if this is eval.
  if mode == tf.estimator.ModeKeys.EVAL:
    num_epochs = 1
    shuffle = False

  y = np.asarray(y).astype("float32").reshape((-1, 1))
  return tf.estimator.inputs.numpy_input_fn(
      x,
      y=y,
      batch_size=batch_size,
      num_epochs=num_epochs,
      shuffle=shuffle,
      queue_capacity=50000)


def serving_input_fn():
  Defines the features to be passed to the model during inference.

After Change

    inputs = (features, labels)

  // Convert the inputs to a Dataset.
  dataset = tf.data.Dataset.from_tensor_slices(inputs)
  if mode == tf.estimator.ModeKeys.TRAIN:
    dataset = dataset.shuffle(1000).repeat().batch(batch_size)
  if mode == tf.estimator.ModeKeys.EVAL:
    dataset = dataset.batch(batch_size)
  return dataset.make_one_shot_iterator().get_next()


def serving_input_fn():
  Defines the features to be passed to the model during inference.

Instances