Use regional images for Kmeans notebooks

Author: winstonaws

sagemaker-python-sdk/1P_kmeans_highlevel/kmeans_mnist.ipynb

@@ -43,7 +43,7 @@
     "\n",
     "Here we set up the linkage and authentication to AWS services. There are three parts to this:\n",
     "\n",
-    "1. The credentials and region for the account that's running training. Upload the credentials in the normal AWS credentials file format using the jupyter upload feature. The region must always be `us-west-2` during the Beta program.\n",
+    "1. The credentials and region for the account that's running training. Upload the credentials in the normal AWS credentials file format using the jupyter upload feature.\n",
     "2. The roles used to give learning and hosting access to your data. See the documentation for how to specify these.\n",
     "3. The S3 bucket that you want to use for training and model data.\n",
     "\n",
@@ -292,9 +292,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import sagemaker\n",
+    "# Uncomment and run to delete\n",
     "\n",
-    "sagemaker.Session().delete_endpoint(kmeans_predictor.endpoint)"
+    "#import sagemaker\n",
+    "#sagemaker.Session().delete_endpoint(kmeans_predictor.endpoint)"
    ]
   },
   {

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

@@ -174,6 +174,15 @@
     "After setting training parameters, we kick off training, and poll for status until training is completed, which in this example, takes between 7 and 11 minutes."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "image"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -187,10 +196,13 @@
     "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",
+    "\n",
     "create_training_params = \\\n",
     "{\n",
     "    \"AlgorithmSpecification\": {\n",
-    "        \"TrainingImage\": \"174872318107.dkr.ecr.us-west-2.amazonaws.com/kmeans:1\",\n",
+    "        \"TrainingImage\": image,\n",
     "        \"TrainingInputMode\": \"File\"\n",
     "    },\n",
     "    \"RoleArn\": role,\n",
@@ -278,7 +290,7 @@
     "model_data = info['ModelArtifacts']['S3ModelArtifacts']\n",
     "\n",
     "primary_container = {\n",
-    "    'Image': \"174872318107.dkr.ecr.us-west-2.amazonaws.com/kmeans:1\",\n",
+    "    'Image': image,\n",
     "    'ModelDataUrl': model_data\n",
     "}\n",
     "\n",