Add the missing cell

Author: gurumurthys

introduction_to_amazon_algorithms/imageclassification_caltech/Image-classification-fulltraining.ipynb

@@ -46,9 +46,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "%%time\n",
@@ -59,7 +57,6 @@
     "role = re.sub(r'^(.+)sts::(\\d+):assumed-role/(.+?)/.*$', r'\\1iam::\\2:role/\\3', assumed_role)\n",
     "\n",
     "bucket='<<bucket-name>>' # customize to your bucket\n",
-    "\n",
     "containers = {'us-west-2': '433757028032.dkr.ecr.us-west-2.amazonaws.com/image-classification:latest',\n",
     "              'us-east-1': '811284229777.dkr.ecr.us-east-1.amazonaws.com/image-classification:latest',\n",
     "              'us-east-2': '825641698319.dkr.ecr.us-east-2.amazonaws.com/image-classification:latest',\n",
@@ -78,9 +75,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import os \n",
@@ -137,9 +132,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "# The algorithm supports multiple network depth (number of layers). They are 18, 34, 50, 101, 152 and 200\n",
@@ -172,7 +165,6 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
-    "collapsed": true,
     "scrolled": true
    },
    "outputs": [],
@@ -254,9 +246,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "# create the Amazon SageMaker training job\n",
@@ -283,9 +273,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "training_info = sagemaker.describe_training_job(TrainingJobName=job_name)\n",
@@ -336,9 +324,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "%%time\n",
@@ -384,24 +370,23 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
-    "%%time\n",
-    "import time\n",
+    "from time import gmtime, strftime\n",
     "\n",
     "timestamp = time.strftime('-%Y-%m-%d-%H-%M-%S', time.gmtime())\n",
-    "endpoint_name = job_name_prefix + '-ep-' + timestamp\n",
-    "print('Endpoint name: {}'.format(endpoint_name))\n",
-    "\n",
-    "endpoint_params = {\n",
-    "    'EndpointName': endpoint_name,\n",
-    "    'EndpointConfigName': endpoint_config_name,\n",
-    "}\n",
-    "endpoint_response = sagemaker.create_endpoint(**endpoint_params)\n",
-    "print('EndpointArn = {}'.format(endpoint_response['EndpointArn']))"
+    "endpoint_config_name = job_name_prefix + '-epc-' + timestamp\n",
+    "endpoint_config_response = sage.create_endpoint_config(\n",
+    "    EndpointConfigName = endpoint_config_name,\n",
+    "    ProductionVariants=[{\n",
+    "        'InstanceType':'ml.p2.xlarge',\n",
+    "        'InitialInstanceCount':3,\n",
+    "        'ModelName':model_name,\n",
+    "        'VariantName':'AllTraffic'}])\n",
+    "\n",
+    "print('Endpoint configuration name: {}'.format(endpoint_config_name))\n",
+    "print('Endpoint configuration arn:  {}'.format(endpoint_config_response['EndpointConfigArn']))"
    ]
   },
   {
@@ -415,9 +400,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "%%time\n",
@@ -445,9 +428,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "# get the status of the endpoint\n",
@@ -493,9 +474,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import boto3\n",
@@ -512,9 +491,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "!wget -O /tmp/test.jpg http://www.vision.caltech.edu/Image_Datasets/Caltech256/images/008.bathtub/008_0007.jpg\n",
@@ -527,9 +504,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import json\n",
@@ -563,9 +538,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "#sage.delete_endpoint(EndpointName=endpoint_name)\n",
@@ -575,9 +548,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": []
   }