commit-live-students · AdityaMV1215 · Jun 1, 2018 · Jun 6, 2018
diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_k_means/__pycache__/__init__.cpython-36.pyc b/q01_k_means/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_k_means/__pycache__/build.cpython-36.pyc b/q01_k_means/__pycache__/build.cpython-36.pyc
diff --git a/q01_k_means/build.py b/q01_k_means/build.py
@@ -1,17 +1,25 @@
 # Default imports
 from sklearn.cluster import KMeans
+import matplotlib
+matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 from sklearn import datasets
+from sklearn.preprocessing import scale
+from sklearn.decomposition import PCA
 
 
 digits = datasets.load_digits()
 
-X_train = digits.images
+X_train = digits.data
 y_train = digits.target
 
 # Write your solution here :
-
-
-
-
-
+def k_means(X_train,y_train,cluster=10,random_state=9):
+    model = KMeans(n_clusters=cluster,random_state=random_state)
+    pca = PCA(n_components=2)
+    reduced_X = pca.fit_transform(X_train)
+    model.fit(reduced_X)
+    ans = model.predict(reduced_X)
+    plt.plot(reduced_X)
+    plt.plot(ans)
+    plt.show()
diff --git a/q01_k_means/tests/__pycache__/__init__.cpython-36.pyc b/q01_k_means/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_k_means/tests/__pycache__/test_q01_k_means.cpython-36.pyc b/q01_k_means/tests/__pycache__/test_q01_k_means.cpython-36.pyc
diff --git a/q02_hierarchy_clustering/__pycache__/__init__.cpython-36.pyc b/q02_hierarchy_clustering/__pycache__/__init__.cpython-36.pyc
diff --git a/q02_hierarchy_clustering/__pycache__/build.cpython-36.pyc b/q02_hierarchy_clustering/__pycache__/build.cpython-36.pyc
diff --git a/q02_hierarchy_clustering/tests/__pycache__/__init__.cpython-36.pyc b/q02_hierarchy_clustering/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q02_hierarchy_clustering/tests/__pycache__/test_q02_hierarchy_clustering.cpython-36.pyc b/q02_hierarchy_clustering/tests/__pycache__/test_q02_hierarchy_clustering.cpython-36.pyc