urban/data_mining_algorithms
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

kmeans Minor Bugfixes

Browse Source 
- Added cluster to test generator
- Added sample data to main
This commit is contained in:
Conrad
2018-06-02 23:30:07 +02:00
parent 7ea392c302
commit 4e3ceac4a9
4 changed files with 62 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/algorithms/kmeansMkI.py
View File

@@ -107,8 +107,8 @@ def assignCluster(data, highPoint, clusters):
# Check the difference between the point (item) and each cluster and set min_cluster to the smallest difference
for cluster in range(0, clusters):
if min_cluster > dmlib.calcdiff(data[item], globals()["cpoint_" + str(cluster)], new_data[0]):
min_cluster = dmlib.calcdiff(data[item], globals()["cpoint_" + str(cluster)], new_data[0])
if min_cluster > dmlib.calcdiff(data[item], globals()["cpoint_" + str(cluster)]):
min_cluster = dmlib.calcdiff(data[item], globals()["cpoint_" + str(cluster)])
assinged_cluster = globals()["cpoint_" + str(cluster)]
# Assign the minimal difference cluster to the data
data_assigned.append(assinged_cluster)
@@ -135,6 +135,6 @@ def startup(data):
print(str(seconds) + " seconds for execution")
# Start the algorithm and generate test data
data = dmtest.plzGen(1000)
data = dmtest.plzGen(10000)
startup(data)