Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def apply_kmeans(data, k):
|
|
27 |
|
28 |
|
29 |
def main():
|
30 |
-
st.title("K-means Clustering
|
31 |
|
32 |
# Global variables declaration
|
33 |
global traffic_df, nature_df, population_df
|
@@ -47,12 +47,12 @@ def main():
|
|
47 |
dataset_mapping = {
|
48 |
"๊ตํต์ ๊ทผ์ฑ": (traffic_df, 'o'),
|
49 |
"์์ฐํ๊ฒฝ": (nature_df, 'x'),
|
50 |
-
"์ธ๊ตฌ๋ฐ์ง๋": (population_df, '
|
51 |
}
|
52 |
-
|
53 |
fig, ax = plt.subplots(figsize=(8, 8))
|
54 |
-
|
55 |
-
|
56 |
data, marker = dataset_mapping[dataset_name]
|
57 |
centroids, labels = apply_kmeans(data.values, k_value)
|
58 |
|
|
|
27 |
|
28 |
|
29 |
def main():
|
30 |
+
st.title("K-means Clustering Simulator")
|
31 |
|
32 |
# Global variables declaration
|
33 |
global traffic_df, nature_df, population_df
|
|
|
47 |
dataset_mapping = {
|
48 |
"๊ตํต์ ๊ทผ์ฑ": (traffic_df, 'o'),
|
49 |
"์์ฐํ๊ฒฝ": (nature_df, 'x'),
|
50 |
+
"์ธ๊ตฌ๋ฐ์ง๋": (population_df, '^')
|
51 |
}
|
52 |
+
|
53 |
fig, ax = plt.subplots(figsize=(8, 8))
|
54 |
+
|
55 |
+
for dataset_name in datasets:
|
56 |
data, marker = dataset_mapping[dataset_name]
|
57 |
centroids, labels = apply_kmeans(data.values, k_value)
|
58 |
|