diff --git a/samples/core/get_started/premade_estimator.py b/samples/core/get_started/premade_estimator.py index 6b771ea3fcd..c4d21ad2015 100644 --- a/samples/core/get_started/premade_estimator.py +++ b/samples/core/get_started/premade_estimator.py @@ -73,9 +73,9 @@ def main(argv): labels=None, batch_size=args.batch_size)) - for pred_dict, expec in zip(predictions, expected): - template = ('\nPrediction is "{}" ({:.1f}%), expected "{}"') + template = ('\nPrediction is "{}" ({:.1f}%), expected "{}"') + for pred_dict, expec in zip(predictions, expected): class_id = pred_dict['class_ids'][0] probability = pred_dict['probabilities'][class_id]