fix experiments sample code

aws-sagemaker/2.experiment_training/keras_experiment.ipynb

@@ -82,11 +82,8 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "import json\n",
         "import boto3\n",
         "import sagemaker\n",
-        "from sagemaker.session import Session\n",
-        "from sagemaker import get_execution_role\n",
         "from sagemaker.experiments.run import Run"
       ]
     },
@@ -121,8 +118,8 @@
         "default_bucket = sagemaker_session.default_bucket()\n",
         "\n",
         "# sagemaker를 사용\n",
-        "sm = sagemaker_session.client(\"sagemaker\")\n",
-        "region = sagemaker_session.region_name"
+        "sm = boto_session.client(\"sagemaker\")\n",
+        "region = boto_session.region_name"
       ]
     },
     {
@@ -192,8 +189,7 @@
         "    train_path,\n",
         ")\n",
         "s3.download_file(\n",
-        "    f\"sagemaker-example-files-prod-{\n",
-        "        region}\", \"datasets/image/MNIST/numpy/input_test.npy\", test_path\n",
+        "    f\"sagemaker-example-files-prod-{region}\", \"datasets/image/MNIST/numpy/input_test.npy\", test_path\n",
         ")\n",
         "s3.download_file(\n",
         "    f\"sagemaker-example-files-prod-{region}\",\n",
@@ -357,9 +353,9 @@
         "# The Keras Callback class provides a method on_epoch_end which emits metrics at the end of each epoch.\n",
         "# All emitted metrics will be logged in the run passed to the callback.\n",
         "class ExperimentCallback(keras.callbacks.Callback):\n",
-        "    def __init__(self, run, model, x_test, y_test):\n",
+        "    def __init__(self, run, trained_model, x_test, y_test):\n",
         "        self.run = run\n",
-        "        self.model = model\n",
+        "        self.trained_model = trained_model\n",
         "        self.x_test = x_test\n",
         "        self.y_test = y_test\n",
         "\n",