Skip to content

Instantly share code, notes, and snippets.

@yufengg
Created November 3, 2017 14:58
Show Gist options
  • Save yufengg/61457707326cc8ade9c0de84597a39de to your computer and use it in GitHub Desktop.
Save yufengg/61457707326cc8ade9c0de84597a39de to your computer and use it in GitHub Desktop.
X = DATA_SETS.test.images[5000:5005]
predict_input_fn =
tf.estimator.inputs.numpy_input_fn(
x={'pixels': X},
batch_size=1,
num_epochs=1,
shuffle=False)
predictions = classifier.predict(
input_fn=predict_input_fn)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment