Updating: adding dbscan

04-dbscan/README.md

@@ -0,0 +1,5 @@
+# DBScan
+
+We will implement DBScan together in class
+
+In addition to the requirements.txt, you need to install the `cs506` python package from [02-library](../02-library/).

04-dbscan/main.py

@@ -0,0 +1,18 @@
+import numpy as np
+import matplotlib.pyplot as plt
+import sklearn.datasets as datasets
+from sklearn.preprocessing import StandardScaler
+
+from cs506 import dbscan
+
+centers = [[1, 1], [-1, -1], [1, -1]]
+X, _ = datasets.make_blobs(n_samples=750, centers=centers, cluster_std=0.4,
+                            random_state=0)
+plt.scatter(X[:,0],X[:,1],s=10, alpha=0.8)
+plt.show()
+
+clustering = dbscan.DBC(X, 3, .2).dbscan()
+colors = np.array([x for x in 'bgrcmykbgrcmykbgrcmykbgrcmyk'])
+colors = np.hstack([colors] * 20)
+plt.scatter(X[:, 0], X[:, 1], color=colors[clustering].tolist(), s=10, alpha=0.8)
+plt.show()

04-dbscan/requirements.txt

@@ -0,0 +1,11 @@
+cycler==0.10.0
+joblib==0.14.0
+kiwisolver==1.1.0
+matplotlib==3.1.1
+numpy==1.17.4
+pyparsing==2.4.5
+python-dateutil==2.8.1
+scikit-learn==0.21.3
+scipy==1.3.2
+six==1.13.0
+sklearn==0.0