fix SKlearn_on_SageMaker_end2end notebook (#1044)

Author: knakad

sagemaker-python-sdk/scikit_learn_randomforest/Sklearn_on_SageMaker_end2end.ipynb

@@ -232,9 +232,9 @@
    "source": [
     "! python script.py --n-estimators 100 \\\n",
     "                   --min-samples-leaf 2 \\\n",
-    "                   --model-dir /home/ec2-user/SageMaker \\\n",
-    "                   --train /home/ec2-user/SageMaker \\\n",
-    "                   --test /home/ec2-user/SageMaker \\\n",
+    "                   --model-dir ./ \\\n",
+    "                   --train ./ \\\n",
+    "                   --test ./ \\\n",
     "                   --features 'CRIM ZN INDUS CHAS NOX RM AGE DIS RAD TAX PTRATIO B LSTAT' \\\n",
     "                   --target target"
    ]
@@ -492,6 +492,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "sklearn_estimator.latest_training_job.wait(logs='None')\n",
     "artifact = sm_boto3.describe_training_job(\n",
     "    TrainingJobName=sklearn_estimator.latest_training_job.name)['ModelArtifacts']['S3ModelArtifacts']\n",
     "\n",