tsne: Print where file is saved to.

This commit is contained in:
Brandon Amos 2015-12-12 16:30:16 -05:00
parent 446436ced1
commit 139cfb7756
1 changed files with 4 additions and 1 deletions

View File

@ -45,4 +45,7 @@ for c, i, target_name in zip(colors,
plt.scatter(X_r[y == i, 0], X_r[y == i, 1], plt.scatter(X_r[y == i, 0], X_r[y == i, 1],
c=c, label=target_name) c=c, label=target_name)
plt.legend() plt.legend()
plt.savefig("{}/tsne.pdf".format(args.workDir))
out = "{}/tsne.pdf".format(args.workDir)
plt.savefig(out)
print("Saved to: {}".format(out))