1.geo.py 2.add test.py 3.readme 4.ideas

Author: houhuawei23

.gitignore

@@ -11,5 +11,9 @@
 **/res
 **/pics
 **/results
-**/test.py
-**.pkl
+**.pkl
+**/resources
+hhw_code/plt_test.py
+hhw_code/tmp.ipynb
+hhw_code/dim_reduce/
+hhw_code/restats

hhw_code/dim_reduce.py

@@ -0,0 +1,19 @@
+import umap
+from sklearn.datasets import fetch_openml
+import matplotlib.pyplot as plt
+import seaborn as sns
+
+sns.set(context="paper", style="white")
+
+mnist = fetch_openml("mnist_784", version=1)
+
+reducer = umap.UMAP(random_state=42)
+embedding = reducer.fit_transform(mnist.data)
+
+fig, ax = plt.subplots(figsize=(12, 10))
+color = mnist.target.astype(int)
+plt.scatter(embedding[:, 0], embedding[:, 1], c=color, cmap="Spectral", s=0.1)
+plt.setp(ax, xticks=[], yticks=[])
+plt.title("MNIST data embedded into two dimensions by UMAP", fontsize=18)
+
+plt.show()