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:
parent
93e2a37a5f
commit
eba4b9ff1e
@ -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))
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user