Updated: kmeans_byom from feedback

Author: djarpin

advanced_functionality/kmeans_bring_your_own_model/kmeans_bring_your_own_model.ipynb

@@ -33,10 +33,12 @@
     "---\n",
     "## Setup\n",
     "\n",
+    "_This notebook was created and tested on an ml.m4.xlarge notebook instance._\n",
+    "\n",
     "Let's start by specifying:\n",
     "\n",
-    "- The IAM role arn used to give learning and hosting access to your data. See the documentation for how to create these.  Note, if more than one role is required for notebook instances, training, and/or hosting, please replace the boto call with a the appropriate full IAM role arn string.\n",
-    "- The S3 bucket and prefix where you'll be storing your model data."
+    "- The S3 bucket and prefix that you want to use for training and model data.  This should be within the same region as the Notebook Instance, training, and hosting.\n",
+    "- The IAM role arn used to give training and hosting access to your data. See the documentation for how to create these.  Note, if more than one role is required for notebook instances, training, and/or hosting, please replace the boto regexp with a the appropriate full IAM role arn string(s)."
    ]
   },
   {
@@ -47,12 +49,15 @@
    },
    "outputs": [],
    "source": [
-    "import boto3\n",
-    "\n",
-    "role = boto3.client('iam').list_instance_profiles()['InstanceProfiles'][0]['Roles'][0]['Arn']\n",
-    "\n",
     "bucket = '<your_s3_bucket_name_here>'\n",
-    "prefix = 'sagemaker/kmeans_byom'"
+    "prefix = 'sagemaker/kmeans_byom'\n",
+    " \n",
+    "# Define IAM role\n",
+    "import boto3\n",
+    "import re\n",
+    " \n",
+    "assumed_role = boto3.client('sts').get_caller_identity()['Arn']\n",
+    "role = re.sub(r'^(.+)sts::(\\d+):assumed-role/(.+?)/.*$', r'\\1iam::\\2:role/\\3', assumed_role)"
    ]
   },
   {
@@ -100,7 +105,7 @@
     "\n",
     "### Data\n",
     "\n",
-    "For simplicity, we'll utilize the MNIST handwritten digit dataset."
+    "For simplicity, we'll utilize the MNIST dataset.  This includes roughly 70K 28 x 28 pixel images of handwritten digits from 0 to 9.  More detail can be found [here](https://en.wikipedia.org/wiki/MNIST_database)."
    ]
   },
   {
@@ -351,10 +356,18 @@
     "\n",
     "This notebook showed how to seed a pre-existing model in an already built container.  This functionality could be replicated with other Amazon SageMaker Algorithms, as well as the TensorFlow and MXNet containers.  Although this is certainly an easy method to bring your own model, it is not likely to provide the flexibility of a bringing your own scoring container.  Please refer to other example notebooks which show how to dockerize your own training and scoring could which could be modified appropriately to your use case."
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sm.delete_endpoint(EndpointName=kmeans_endpoint)"
+   ]
   }
  ],
  "metadata": {
-  "notice": "Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.  Licensed under the Apache License, Version 2.0 (the \"License\"). You may not use this file except in compliance with the License. A copy of the License is located at http://aws.amazon.com/apache2.0/ or in the \"license\" file accompanying this file. This file is distributed on an \"AS IS\" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.",
   "kernelspec": {
    "display_name": "Environment (conda_mxnet_p36)",
    "language": "python",
@@ -371,7 +384,8 @@
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
    "version": "3.6.3"
-  }
+  },
+  "notice": "Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.  Licensed under the Apache License, Version 2.0 (the \"License\"). You may not use this file except in compliance with the License. A copy of the License is located at http://aws.amazon.com/apache2.0/ or in the \"license\" file accompanying this file. This file is distributed on an \"AS IS\" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License."
  },
  "nbformat": 4,
  "nbformat_minor": 2