diff options
author | nunzip <np.scarh@gmail.com> | 2019-03-10 18:15:56 +0000 |
---|---|---|
committer | nunzip <np.scarh@gmail.com> | 2019-03-10 18:15:56 +0000 |
commit | 8a8c3e9f19ea0114793edba4fce4a73e2624f6a5 (patch) | |
tree | 451809dd25c49b179bdce3f5a156ac0123c74718 | |
parent | 4f96cbcc673f46d8fdd5bc7bc253aaeb2aa2af88 (diff) | |
parent | 8a3c9ecfec05ce73f0fe69426bfc97110b6f189f (diff) | |
download | e4-gan-8a8c3e9f19ea0114793edba4fce4a73e2624f6a5.tar.gz e4-gan-8a8c3e9f19ea0114793edba4fce4a73e2624f6a5.tar.bz2 e4-gan-8a8c3e9f19ea0114793edba4fce4a73e2624f6a5.zip |
Merge branch 'master' of skozl.com:e4-gan
-rw-r--r-- | lenet.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -161,7 +161,7 @@ def test_classifier(model, x_test, y_true, conf_mat=False, pca=False, tsne=False set_pca = PCA(n_components=2) pca_rep = set_pca.fit_transform(logits) pca_rep, y_tmp = shuffle(pca_rep, y_true, random_state=0) - plt.scatter(pca_rep[:1000, 0], pca_rep[:1000, 1], c=y_true[:1000], edgecolor='none', alpha=0.5, cmap=plt.cm.get_cmap('Paired', 10)) + plt.scatter(pca_rep[:1000, 0], pca_rep[:1000, 1], c=y_tmp[:1000], edgecolor='none', alpha=0.5, cmap=plt.cm.get_cmap('Paired', 10)) plt.xlabel('component 1') plt.ylabel('component 2') plt.colorbar(); @@ -171,7 +171,7 @@ def test_classifier(model, x_test, y_true, conf_mat=False, pca=False, tsne=False components = tsne.fit_transform(logits) print(components.shape) components, y_tmp = shuffle(components, y_true, random_state=0) - plt.scatter(components[:1000, 0], components[:1000, 1], c=y_true[:1000], edgecolor='none', alpha=0.5, cmap=plt.cm.get_cmap('Paired', 10)) + plt.scatter(components[:1000, 0], components[:1000, 1], c=y_tmp[:1000], edgecolor='none', alpha=0.5, cmap=plt.cm.get_cmap('Paired', 10)) plt.xlabel('component 1') plt.ylabel('component 2') plt.colorbar(); |