Change PCA color

1 files changed, 1 insertions, 1 deletions

diff --git a/lenet.py b/lenet.py
index 7c4c48c..39a5cd7 100644
--- a/lenet.py
+++ b/lenet.py
@@ -176,7 +176,7 @@ def test_classifier(model, x_test, y_true, conf_mat=False, pca=False):
     pca_rep = set_pca.fit_transform(pca_rep)
     print(pca_rep.shape)
     pca_rep, y_tmp = shuffle(pca_rep, y_true, random_state=0)
-    plt.scatter(pca_rep[:100, 0], pca_rep[:100, 1], c=y_true[:100], edgecolor='none', alpha=0.5, cmap=plt.cm.get_cmap('Spectral', 10))
+    plt.scatter(pca_rep[:100, 0], pca_rep[:100, 1], c=y_true[:100], edgecolor='none', alpha=0.5, cmap=plt.cm.get_cmap('Paired', 10))
     plt.xlabel('component 1')
     plt.ylabel('component 2')
     plt.colorbar();