diff --git a/__init__.pyc b/__init__.pyc index 6c0d88a..6a03976 100644 Binary files a/__init__.pyc and b/__init__.pyc differ diff --git a/q01_k_means/__init__.pyc b/q01_k_means/__init__.pyc index bff55bc..e3878dd 100644 Binary files a/q01_k_means/__init__.pyc and b/q01_k_means/__init__.pyc differ diff --git a/q01_k_means/build.py b/q01_k_means/build.py index fca565c..1296a65 100644 --- a/q01_k_means/build.py +++ b/q01_k_means/build.py @@ -1,4 +1,3 @@ -# Default imports from sklearn.cluster import KMeans import matplotlib.pyplot as plt from sklearn import datasets @@ -9,7 +8,29 @@ X_train = digits.images y_train = digits.target -# Write your solution here : + +def k_means(X_train, y_train, cluster=10, random_state=9): + X = X_train.reshape((len(X_train), -1)) + kmeans = KMeans(n_clusters=cluster, random_state=random_state).fit(X, y_train) + a = X_train[(y_train == 0) & (kmeans.labels_ == 0)][0:20] + b = X_train[(y_train == 1) & (kmeans.labels_ == 1)][0:20] + c = X_train[(y_train == 2) & (kmeans.labels_ == 2)][0:20] + d = X_train[(y_train == 3) & (kmeans.labels_ == 3)][0:20] + e = X_train[(y_train == 4) & (kmeans.labels_ == 4)][0:20] + f = X_train[(y_train == 5) & (kmeans.labels_ == 5)][0:20] + g = X_train[(y_train == 6) & (kmeans.labels_ == 6)][0:20] + h = X_train[(y_train == 7) & (kmeans.labels_ == 7)][0:20] + i = X_train[(y_train == 8) & (kmeans.labels_ == 8)][0:20] + j = X_train[(y_train == 9) & (kmeans.labels_ == 9)][0:20] + for char in (a, b, c, d, e, f, g, h, i, j): + for index in range(0, len(char)): + plt.subplot(10, 20, index + 1) + plt.axis('off') + plt.imshow(char[index]) + plt.show() + + + diff --git a/q01_k_means/build.pyc b/q01_k_means/build.pyc index fa56657..e77631e 100644 Binary files a/q01_k_means/build.pyc and b/q01_k_means/build.pyc differ diff --git a/q01_k_means/tests/__init__.pyc b/q01_k_means/tests/__init__.pyc index f6a37b9..ed0e1a7 100644 Binary files a/q01_k_means/tests/__init__.pyc and b/q01_k_means/tests/__init__.pyc differ diff --git a/q01_k_means/tests/test_q01_k_means.pyc b/q01_k_means/tests/test_q01_k_means.pyc index ac55928..a836cbe 100644 Binary files a/q01_k_means/tests/test_q01_k_means.pyc and b/q01_k_means/tests/test_q01_k_means.pyc differ