Skip to content

Instantly share code, notes, and snippets.

@UrusuLambda

UrusuLambda/umap-2d-3d-data.py

Last active August 14, 2020 17:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save UrusuLambda/f24da246a11dcbf591be11b7f02b29d2 to your computer and use it in GitHub Desktop.
Save UrusuLambda/f24da246a11dcbf591be11b7f02b29d2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
umap-2d-3d-data.py
#For 2D
x_train_tranformed_2d = umap.UMAP(n_neighbors=5, n_components=2).fit(x_train).embedding_
x_test_tranformed_2d = umap.UMAP(n_neighbors=5, n_components=2).fit(x_test).embedding_
#For 3D
x_train_tranformed_3d = umap.UMAP(n_neighbors=5, n_components=3).fit(x_train).embedding_
x_test_tranformed_3d = umap.UMAP(n_neighbors=5, n_components=3).fit(x_test).embedding_
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment