Fixed predictions showing as array of False instead of a single True or False

eitansela · eitansela · commit 1c68e98aeda2 · 2020-11-02T12:44:55.000+02:00
diff --git a/sagemaker-python-sdk/tensorflow_script_mode_training_and_serving/tensorflow_script_mode_training_and_serving.ipynb b/sagemaker-python-sdk/tensorflow_script_mode_training_and_serving/tensorflow_script_mode_training_and_serving.ipynb
@@ -268,7 +268,7 @@
    "source": [
     "predictions = predictor.predict(train_data[:50])\n",
     "for i in range(0, 50):\n",
-    "    prediction = np.argmax(predictions['predictions'][i])\n",
+    "    prediction = predictions['predictions'][i]['classes']\n",
     "    label = train_labels[i]\n",
     "    print('prediction is {}, label is {}, matched: {}'.format(prediction, label, prediction == label))"
    ]

Original file line number	Diff line number	Diff line change
`@@ -268,7 +268,7 @@`
`268`	`268`	`"source": [`
`269`	`269`	`"predictions = predictor.predict(train_data[:50])\n",`
`270`	`270`	`"for i in range(0, 50):\n",`
`271`		`- " prediction = np.argmax(predictions['predictions'][i])\n",`
	`271`	`+ " prediction = predictions['predictions'][i]['classes']\n",`
`272`	`272`	`" label = train_labels[i]\n",`
`273`	`273`	`" print('prediction is {}, label is {}, matched: {}'.format(prediction, label, prediction == label))"`
`274`	`274`	`]`