[LDA] Add sagemaker_session to training jobs

benjamin-fenker · Nov 28, 2017 · 205c4a6 · 205c4a6
1 parent 22f1a35
commit 205c4a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/introduction_to_amazon_algorithms/lda_topic_modeling/LDA-Introduction.ipynb b/introduction_to_amazon_algorithms/lda_topic_modeling/LDA-Introduction.ipynb
@@ -342,6 +342,7 @@
     "    output_path='s3://{}/{}/output'.format(bucket, prefix),\n",
     "    train_instance_count=1,\n",
     "    train_instance_type='ml.c4.2xlarge',\n",
+    "    sagemaker_session=session,\n",
     ")\n",
     "\n",
     "# set algorithm-specific hyperparameters\n",

diff --git a/scientific_details_of_algorithms/lda_topic_modeling/LDA-Science.ipynb b/scientific_details_of_algorithms/lda_topic_modeling/LDA-Science.ipynb
@@ -584,6 +584,7 @@
     "    output_path='s3://{}/{}/output'.format(bucket, prefix),\n",
     "    train_instance_count=1,\n",
     "    train_instance_type='ml.c4.2xlarge',\n",
+    "    sagemaker_session=session,\n",
     ")\n",
     "\n",
     "# set algorithm-specific hyperparameters\n",