import umap from sklearn.datasets import load_digits from scipy.sparse.csgraph import connected_components import matplotlib.pyplot as plt import matplotlib.cm as cm from sklearn.manifold import TSNE import time def main(): digits = load_digits() digits.target = [float(digits.target[i]) for i in range(len(digits.target))] # UMAP start_time = time.time() embedding = umap.UMAP().fit_transform(digits