Increase number of training images in demo

microsoft · Nov 29, 2018 · 6654cac · 6654cac
1 parent d870e8e
commit 6654cac
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/notebooks/samples/ModelInterpretation - Snow Leopard Detection.ipynb b/notebooks/samples/ModelInterpretation - Snow Leopard Detection.ipynb
@@ -55,7 +55,7 @@
     "        .setOutputCol(\"images\")) \\\n",
     "    .mlTransform(BingImageSearch.getUrlTransformer(\"images\", \"urls\")) \\\n",
     "    .withColumn(\"labels\", lit(name)) \\\n",
-    "    .limit(200)\n"
+    "    .limit(400)\n"
    ]
   },
   {
@@ -169,7 +169,7 @@
     "    .setOutputCol(\"images\")) \\\n",
     "  .mlTransform(BingImageSearch.getUrlTransformer(\"images\", \"urls\")) \\\n",
     "  .withColumn(\"label\", lit(\"other\")) \\\n",
-    "  .limit(200)\n",
+    "  .limit(400)\n",
     "  \n",
     "displayDF(randomLinks)"
    ]
@@ -209,7 +209,7 @@
     "model = Pipeline(stages=[\n",
     "  StringIndexer(inputCol = \"labels\", outputCol=\"index\"),\n",
     "  ImageFeaturizer(inputCol=\"image\", outputCol=\"features\", cutOutputLayers=2).setModel(network),\n",
-    "  LogisticRegression(maxIter=5, labelCol=\"index\", regParam=0.3),\n",
+    "  LogisticRegression(maxIter=5, labelCol=\"index\", regParam=5.0),\n",
     "  UDFTransformer()\\\n",
     "      .setUDF(udf(getIndex, DoubleType()))\\\n",
     "      .setInputCol(\"probability\")\\\n",
@@ -266,7 +266,7 @@
     "  .setModelPartitions(50)\\\n",
     "  .setCellSize(100.0)\\\n",
     "  .setModifier(50.0)\\\n",
-    "  .setNSamples(200)\n",
+    "  .setNSamples(300)\n",
     "\n",
     "result = lime.transform(test_subsample)"
    ]