Merge pull request aws#417 from asadoughi/master

Update knn notebook to use SDK-based uri

Author: djarpin

introduction_to_amazon_algorithms/k_nearest_neighbors_covtype/k_nearest_neighbors_covtype.ipynb

@@ -184,6 +184,7 @@
     "import sagemaker\n",
     "from sagemaker import get_execution_role\n",
     "from sagemaker.predictor import csv_serializer, json_deserializer\n",
+    "from sagemaker.amazon.amazon_estimator import get_image_uri\n",
     "\n",
     "\n",
     "def trained_estimator_from_hyperparams(s3_train_data, hyperparams, output_path, s3_test_data=None):\n",
@@ -192,18 +193,8 @@
     "    and return a deployed predictor\n",
     "    \n",
     "    \"\"\"\n",
-    "    # specify algorithm containers. These contain the code for the training job\n",
-    "    containers = {\n",
-    "        'us-west-2': '174872318107.dkr.ecr.us-west-2.amazonaws.com/knn:1',\n",
-    "        'us-east-1': '382416733822.dkr.ecr.us-east-1.amazonaws.com/knn:1',\n",
-    "        'us-east-2': '404615174143.dkr.ecr.us-east-2.amazonaws.com/knn:1',\n",
-    "        'eu-west-1': '438346466558.dkr.ecr.eu-west-1.amazonaws.com/knn:1',\n",
-    "        'ap-northeast-1': '351501993468.dkr.ecr.ap-northeast-1.amazonaws.com/knn:1',\n",
-    "        'ap-northeast-2': '835164637446.dkr.ecr.ap-northeast-2.amazonaws.com/knn:1',\n",
-    "        'ap-southeast-2': '712309505854.dkr.ecr.ap-southeast-2.amazonaws.com/knn:1'\n",
-    "    }\n",
     "    # set up the estimator\n",
-    "    knn = sagemaker.estimator.Estimator(containers[boto3.Session().region_name],\n",
+    "    knn = sagemaker.estimator.Estimator(get_image_uri(boto3.Session().region_name, \"knn\"),\n",
     "        get_execution_role(),\n",
     "        train_instance_count=1,\n",
     "        train_instance_type='ml.m5.2xlarge',\n",
@@ -397,25 +388,18 @@
     "import sagemaker\n",
     "from sagemaker import get_execution_role\n",
     "from sagemaker.predictor import csv_serializer, json_deserializer\n",
+    "from sagemaker.amazon.amazon_estimator import get_image_uri\n",
+    "\n",
     "\n",
     "def trained_estimator_from_hyperparams(s3_train_data, hyperparams, output_path, s3_test_data=None):\n",
     "    \"\"\"\n",
     "    Create an Estimator from the given hyperparams, fit to training data, \n",
     "    and return a deployed predictor\n",
     "    \n",
     "    \"\"\"\n",
-    "    # specify algorithm containers. These contain the code for the training job\n",
-    "    containers = {\n",
-    "        'us-west-2': '174872318107.dkr.ecr.us-west-2.amazonaws.com/knn:1',\n",
-    "        'us-east-1': '382416733822.dkr.ecr.us-east-1.amazonaws.com/knn:1',\n",
-    "        'us-east-2': '404615174143.dkr.ecr.us-east-2.amazonaws.com/knn:1',\n",
-    "        'eu-west-1': '438346466558.dkr.ecr.eu-west-1.amazonaws.com/knn:1',\n",
-    "        'ap-northeast-1': '351501993468.dkr.ecr.ap-northeast-1.amazonaws.com/knn:1',\n",
-    "        'ap-northeast-2': '835164637446.dkr.ecr.ap-northeast-2.amazonaws.com/knn:1',\n",
-    "        'ap-southeast-2': '712309505854.dkr.ecr.ap-southeast-2.amazonaws.com/knn:1'\n",
-    "    }\n",
+    "\n",
     "    # set up the estimator\n",
-    "    knn = sagemaker.estimator.Estimator(containers[boto3.Session().region_name],\n",
+    "    knn = sagemaker.estimator.Estimator(get_image_uri(boto3.Session().region_name, \"knn\"),\n",
     "        get_execution_role(),\n",
     "        train_instance_count=1,\n",
     "        train_instance_type='ml.m5.2xlarge',\n",
@@ -821,9 +805,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "conda_python3",
    "language": "python",
-   "name": "python3"
+   "name": "conda_python3"
   },
   "language_info": {
    "codemirror_mode": {