From 1e7ef95bbb67c3f7e04dc68ec048ae43361ccace Mon Sep 17 00:00:00 2001 From: Dillon Daudert Date: Thu, 17 Jan 2019 09:16:48 -0500 Subject: [PATCH] Use same n_epochs in notebook --- PlotMNIST.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/PlotMNIST.ipynb b/PlotMNIST.ipynb index 71b40f6..e85ed71 100644 --- 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)))" ]