mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-01-18 16:27:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update dbscan.py with more test cases

Browse Source
This commit is contained in:
tkgowtham 2024-10-02 21:14:43 +05:30 committed by GitHub
parent 94caecf36e
commit 36700a13ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 72 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
machine_learning/dbscan.py
View File

@ -7,6 +7,7 @@ LinkedIn : https://www.linkedin.com/in/gowtham-kamalasekar/
import math import math
import matplotlib.patches as mpatches
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import pandas as pd import pandas as pd
@ -118,6 +119,38 @@ class DbScan:
11 [2, 10, 11, 12] 11 [2, 10, 11, 12]
12 [9, 11, 12] 12 [9, 11, 12]
>>> result = DbScan(3, 2.5).perform_dbscan()
>>> for key in sorted(result):
... print(key, sorted(result[key]))
1 [1, 2, 10, 11]
2 [1, 2, 3, 10, 11]
3 [2, 3, 4, 11]
4 [3, 4, 5, 6, 7, 8]
5 [4, 5, 6, 7, 8]
6 [4, 5, 6, 7]
7 [4, 5, 6, 7, 8]
8 [4, 5, 7, 8]
9 [9, 11, 12]
10 [1, 2, 10, 11, 12]
11 [1, 2, 3, 9, 10, 11, 12]
12 [9, 10, 11, 12]
>>> result = DbScan(5, 2.5).perform_dbscan()
>>> for key in sorted(result):
... print(key, sorted(result[key]))
1 [1, 2, 10, 11]
2 [1, 2, 3, 10, 11]
3 [2, 3, 4, 11]
4 [3, 4, 5, 6, 7, 8]
5 [4, 5, 6, 7, 8]
6 [4, 5, 6, 7]
7 [4, 5, 6, 7, 8]
8 [4, 5, 7, 8]
9 [9, 11, 12]
10 [1, 2, 10, 11, 12]
11 [1, 2, 3, 9, 10, 11, 12]
12 [9, 10, 11, 12]
""" """
if type(self.file) is str: if type(self.file) is str:
data = pd.read_csv(self.file) data = pd.read_csv(self.file)
@ -159,6 +192,35 @@ class DbScan:
10 [1, 10, 11] ---> Noise ---> Border 10 [1, 10, 11] ---> Noise ---> Border
11 [2, 10, 11, 12] ---> Core 11 [2, 10, 11, 12] ---> Core
12 [9, 11, 12] ---> Noise ---> Border 12 [9, 11, 12] ---> Noise ---> Border
>>> DbScan(5,2.5).print_dbscan()
1 [1, 2, 10, 11] ---> Noise ---> Border
2 [1, 2, 3, 10, 11] ---> Core
3 [2, 3, 4, 11] ---> Noise ---> Border
4 [3, 4, 5, 6, 7, 8] ---> Core
5 [4, 5, 6, 7, 8] ---> Core
6 [4, 5, 6, 7] ---> Noise ---> Border
7 [4, 5, 6, 7, 8] ---> Core
8 [4, 5, 7, 8] ---> Noise ---> Border
9 [9, 11, 12] ---> Noise ---> Border
10 [1, 2, 10, 11, 12] ---> Core
11 [1, 2, 3, 9, 10, 11, 12] ---> Core
12 [9, 10, 11, 12] ---> Noise ---> Border
>>> DbScan(2,0.5).print_dbscan()
1 [1] ---> Noise
2 [2] ---> Noise
3 [3] ---> Noise
4 [4] ---> Noise
5 [5] ---> Noise
6 [6] ---> Noise
7 [7] ---> Noise
8 [8] ---> Noise
9 [9] ---> Noise
10 [10] ---> Noise
11 [11] ---> Noise
12 [12] ---> Noise
""" """
for i in self.dict1: for i in self.dict1:
print(i, " ", self.dict1[i], end=" ---> ") print(i, " ", self.dict1[i], end=" ---> ")
@ -185,6 +247,13 @@ class DbScan:
>>> DbScan(4,1.9).plot_dbscan() >>> DbScan(4,1.9).plot_dbscan()
Plotted Successfully Plotted Successfully
>>> DbScan(5,2.5).plot_dbscan()
Plotted Successfully
>>> DbScan(5,2.5).plot_dbscan()
Plotted Successfully
""" """
if type(self.file) is str: if type(self.file) is str:
data = pd.read_csv(self.file) data = pd.read_csv(self.file)
@ -214,10 +283,12 @@ class DbScan:
ha="center", ha="center",
va="bottom", va="bottom",
) )
core_legend = mpatches.Patch(color="red", label="Core")
noise_legend = mpatches.Patch(color="green", label="Noise")
plt.xlabel("X") plt.xlabel("X")
plt.ylabel("Y") plt.ylabel("Y")
plt.title("DBSCAN Clustering") plt.title("DBSCAN Clustering")
plt.legend(["Core", "Noise"]) plt.legend(handles=[core_legend, noise_legend])
plt.show() plt.show()
print("Plotted Successfully") print("Plotted Successfully")