Fix image to use different account per region

Author: winstonaws

sagemaker-python-sdk/1P_kmeans_lowlevel/kmeans_mnist_lowlevel.ipynb

@@ -187,8 +187,11 @@
     "job_name = 'kmeans-lowlevel-' + strftime(\"%Y-%m-%d-%H-%M-%S\", gmtime())\n",
     "print(\"Training job\", job_name)\n",
     "\n",
-    "region = !aws configure get region\n",
-    "image = '174872318107.dkr.ecr.{}.amazonaws.com/kmeans:1'.format(region[0])\n",
+    "images = {'us-west-2': '174872318107.dkr.ecr.us-west-2.amazonaws.com/kmeans:latest',\n",
+    "          'us-east-1': '382416733822.dkr.ecr.us-east-1.amazonaws.com/kmeans:latest',\n",
+    "          'us-east-2': '404615174143.dkr.ecr.us-east-2.amazonaws.com/kmeans:latest',\n",
+    "          'eu-west-1': '438346466558.dkr.ecr.eu-west-1.amazonaws.com/kmeans:latest'}\n",
+    "image = images[boto3.Session().region_name]\n",
     "\n",
     "create_training_params = \\\n",
     "{\n",