Skip to content

Instantly share code, notes, and snippets.

@GeorgeSeif

GeorgeSeif/setup_sklearn_cluster.py

Created January 30, 2021 17:46
Show Gist options
  • Save GeorgeSeif/ed08129da42d5360ae55bb5e4514ae4d to your computer and use it in GitHub Desktop.
Save GeorgeSeif/ed08129da42d5360ae55bb5e4514ae4d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
setup_sklearn_cluster.py
import numpy as np
import time
from sklearn.cluster import KMeans
from keras.datasets import mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train = x_train.reshape(len(x_train), -1).astype(float) / 255.
x_test = x_test.reshape(len(x_test), -1).astype(float) / 255.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment