Use same n_epochs in notebook

dillondaudert · Jan 17, 2019 · 1e7ef95 · 1e7ef95
1 parent 4ee0f0b
commit 1e7ef95
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/PlotMNIST.ipynb b/PlotMNIST.ipynb
@@ -29,7 +29,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "@time res_jl = umap(mnist_x; n_neighbors=10, min_dist=0.001)\n",
+    "@time res_jl = umap(mnist_x; n_neighbors=10, min_dist=0.001, n_epochs=200)\n",
     "\n",
     "scatter(res_jl[1,:], res_jl[2,:], zcolor=mnist_y, \n",
     "        title=\"MNIST: Julia UMAP\", marker=(2, 2, :auto, stroke(0)))"
@@ -41,7 +41,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "@time res_py = py_umap.UMAP(n_neighbors=10, min_dist=0.001)[:fit_transform](permutedims(mnist_x))\n",
+    "@time res_py = py_umap.UMAP(n_neighbors=10, min_dist=0.001, n_epochs=200)[:fit_transform](permutedims(mnist_x))\n",
     "scatter(res_py[:,1], res_py[:,2], zcolor=mnist_y, \n",
     "        title=\"MNIST: Python UMAP\", marker=(2, 2, :auto, stroke(0)))"
    ]
@@ -62,7 +62,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "@time res_jl = umap(fmnist_x; n_neighbors=5, min_dist=0.1)\n",
+    "@time res_jl = umap(fmnist_x; n_neighbors=5, min_dist=0.1, n_epochs=200)\n",
     "x2 = res_jl[1,:]\n",
     "y2 = res_jl[2,:]\n",
     "scatter(res_jl[1,:], res_jl[2,:], zcolor=fmnist_y, \n",
@@ -75,7 +75,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "@time res_py = py_umap.UMAP(n_neighbors=5, min_dist=0.1)[:fit_transform](permutedims(fmnist_x))\n",
+    "@time res_py = py_umap.UMAP(n_neighbors=5, min_dist=0.1, n_epochs=200)[:fit_transform](permutedims(fmnist_x))\n",
     "scatter(res_py[:,1], res_py[:,2], zcolor=fmnist_y, \n",
     "        title=\"FMNIST: Python UMAP\", marker=(2, 2, :auto, stroke(0)))"
    ]