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

kmeans ++ Update

Browse Source 
- Added concept for kmeans ++
This commit is contained in:
Conrad
2018-06-02 23:31:43 +02:00
parent 4e3ceac4a9
commit 25fa068df9
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/algorithms/kmeansMkI.py
View File

@@ -31,9 +31,15 @@ import dmtest
# CODE # CODE
# Main function of the algorithm # Main function of the algorithm
def kmeansmk1(data, clusters): def kmeansmk1(data, clusters):
globals()["cpoint_0"] = data[randint(0, len(data))]
globals()["cpoint_1"] = dmlib.pp_calcdiff(data, globals()["cpoint_0"])
print("Initial cluster 1: " + str(globals()["cpoint_0"]))
print("Initial cluster 2: " + str(globals()["cpoint_1"]))
# Defining cluster points # Defining cluster points
for i in range(0, clusters): for i in range(2, clusters):
globals()["cpoint_" + str(i)] = data[randint(0, len(data))] globals()["cpoint_" + str(i)] = dmlib.pp_calcdiff_2(data, globals()["cpoint_" + str(i - 1)], globals()["cpoint_" + str(i - 2)])
print("Initial cluster " + str(i + 1) + ": " + str(globals()["cpoint_" + str(i)])) print("Initial cluster " + str(i + 1) + ": " + str(globals()["cpoint_" + str(i)]))
# Get max value in the data array # Get max value in the data array