Add comment clarifying items in predictions list

This way, it becomes clear what the result of the `print()` call will look like.
This commit is contained in:
Joppe Geluykens 2018-09-13 00:21:39 +02:00 committed by GitHub
parent 93e2a37a5f
commit eba4b9ff1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -79,6 +79,8 @@ def main(_):
return predict_features
prediction_iterable = estimator.predict(input_fn=predict_input_fn)
# Create a list containing each of the prediction dictionaries that map
# the key 'logits' to an array of model outputs.
predictions = [prediction_iterable.next() for _ in range(10)]
print("Prediction results: {}".format(predictions))