fixes for docker notebooks

microsoft · Jun 28, 2018 · 05c5e96 · 05c5e96
1 parent 5c0e406
commit 05c5e96
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/notebooks/samples/304 - Medical Entity Extraction.ipynb b/notebooks/samples/304 - Medical Entity Extraction.ipynb
@@ -80,12 +80,11 @@
    "outputs": [],
    "source": [
     "modelName = \"BiLSTM\"\n",
-    "modelDir = \"models\"\n",
+    "modelDir = abspath(\"models\")\n",
     "if not os.path.exists(modelDir): os.makedirs(modelDir)\n",
     "d = ModelDownloader(spark, \"file://\" + modelDir)\n",
     "modelSchema = d.downloadByName(modelName)\n",
-    "nltk.download(\"punkt\", download_dir=modelDir)\n",
-    "nltk.data.path.append(modelDir)"
+    "nltk.download(\"punkt\")"
    ]
   },
   {

diff --git a/tools/docker/bin/launcher b/tools/docker/bin/launcher
@@ -23,4 +23,4 @@ fi
 
 PYSPARK_DRIVER_PYTHON="jupyter" \
 PYSPARK_DRIVER_PYTHON_OPTS="notebook --no-browser --port=$port --ip=*" \
-  pyspark --master "local[*]" --repositories "$MML_M2REPOS" --packages "$MML_PACKAGE"
+  pyspark --master "local[2]" --driver-memory 2g --executor-memory 4g --repositories "$MML_M2REPOS" --packages "$MML_PACKAGE"