From 80fdc264a691d6e4efc23394ab64deb1eb3f8621 Mon Sep 17 00:00:00 2001 From: yuanx749 Date: Thu, 11 Nov 2021 18:43:02 +0100 Subject: [PATCH] Accelerate plot_t_sne_perplexity.py --- examples/manifold/plot_t_sne_perplexity.py | 27 ++++++++++++++++++---- 1 file changed, 22 insertions(+), 5 deletions(-) diff --git a/examples/manifold/plot_t_sne_perplexity.py b/examples/manifold/plot_t_sne_perplexity.py index 6fd83f48bdb3f..20fd00a835620 100644 --- a/examples/manifold/plot_t_sne_perplexity.py +++ b/examples/manifold/plot_t_sne_perplexity.py @@ -34,12 +34,14 @@ from sklearn import manifold, datasets from time import time -n_samples = 300 +n_samples = 150 n_components = 2 (fig, subplots) = plt.subplots(3, 5, figsize=(15, 8)) perplexities = [5, 30, 50, 100] -X, y = datasets.make_circles(n_samples=n_samples, factor=0.5, noise=0.05) +X, y = datasets.make_circles( + n_samples=n_samples, factor=0.5, noise=0.05, random_state=0 +) red = y == 0 green = y == 1 @@ -56,7 +58,12 @@ t0 = time() tsne = manifold.TSNE( - n_components=n_components, init="random", random_state=0, perplexity=perplexity + n_components=n_components, + init="random", + random_state=0, + perplexity=perplexity, + learning_rate="auto", + n_iter=300, ) Y = tsne.fit_transform(X) t1 = time() @@ -81,7 +88,12 @@ t0 = time() tsne = manifold.TSNE( - n_components=n_components, init="random", random_state=0, perplexity=perplexity + n_components=n_components, + init="random", + random_state=0, + perplexity=perplexity, + learning_rate="auto", + n_iter=300, ) Y = tsne.fit_transform(X) t1 = time() @@ -114,7 +126,12 @@ t0 = time() tsne = manifold.TSNE( - n_components=n_components, init="random", random_state=0, perplexity=perplexity + n_components=n_components, + init="random", + random_state=0, + perplexity=perplexity, + learning_rate="auto", + n_iter=400, ) Y = tsne.fit_transform(X) t1 = time()