UMAP (Uniform Manifold Approximation and Projection)
Last updated
Last updated
import umap
from sklearn.datasets import load_digits
import matplotlib.pyplot as plt
# 載入資料
X, y = load_digits(return_X_y=True)
# 建立模型
reducer = umap.UMAP(n_neighbors=15, min_dist=0.1, n_components=2, metric='euclidean')
X_embedded = reducer.fit_transform(X)
# 視覺化
plt.scatter(X_embedded[:, 0], X_embedded[:, 1], c=y, cmap='Spectral', s=5)
plt.title("UMAP projection of Digits dataset")
plt.show()