15 lines
436 B
Python
15 lines
436 B
Python
import pandas as pd
|
|
import dbscan_lab_helper as helper
|
|
from sklearn import cluster
|
|
|
|
dataset_1 = pd.read_csv('blobs.csv')[:80].values
|
|
helper.plot_dataset(dataset_1)
|
|
|
|
# TODO: create an instance of DBSCAN
|
|
dbscan = cluster.DBSCAN()
|
|
# TODO: use DBSCAN's fit_predict to return clustering labels for dataset_1
|
|
clustering_labels_1 = dbscan.fit_predict(dataset_1)
|
|
|
|
# Plot clustering
|
|
helper.plot_clustered_dataset(dataset_1, clustering_labels_1)
|