Fix prediction data in tensorflow keras notebook (aws#462)

Author: yangaws

sagemaker-python-sdk/tensorflow_keras_cifar10/cifar10_cnn.py

@@ -90,7 +90,7 @@ def serving_input_fn(hyperparameters):
     # Notice that the input placeholder has the same input shape as the Keras model input
     tensor = tf.placeholder(tf.float32, shape=[None, HEIGHT, WIDTH, DEPTH])
 
-    # The inputs key PREDICT_INPUTS matches the Keras InputLayer name
+    # The inputs key INPUT_TENSOR_NAME matches the Keras InputLayer name
     inputs = {INPUT_TENSOR_NAME: tensor}
     return tf.estimator.export.ServingInputReceiver(inputs, inputs)
 

sagemaker-python-sdk/tensorflow_keras_cifar10/tensorflow_keras_CIFAR10.ipynb

@@ -186,8 +186,8 @@
     "    # Notice that the input placeholder has the same input shape as the Keras model input\n",
     "    tensor = tf.placeholder(tf.float32, shape=[None, HEIGHT, WIDTH, DEPTH])\n",
     "    \n",
-    "    # The inputs key PREDICT_INPUTS matches the Keras InputLayer name\n",
-    "    inputs = {PREDICT_INPUTS: tensor}\n",
+    "    # The inputs key INPUT_TENSOR_NAME matches the Keras InputLayer name\n",
+    "    inputs = {INPUT_TENSOR_NAME: tensor}\n",
     "    return tf.estimator.export.ServingInputReceiver(inputs, inputs)\n",
     "\n",
     "\n",
@@ -269,7 +269,8 @@
     "import numpy as np\n",
     "data = np.random.randn(1, 32, 32, 3)\n",
     "\n",
-    "predictor.predict(data)"
+    "# The inputs key 'inputs_input' matches the Keras InputLayer name\n",
+    "predictor.predict({'inputs_input': data}) "
    ]
   },
   {