fix: update n_init of kmeans

Author: galogm

.ci/install-dev.sh

@@ -1,6 +1,6 @@
 rm -rf .env
 
-# # install python 3.8.16 use pyenv
+# # install python 3.8.16 using pyenv:
 # pyenv install 3.8.16
 # pyenv local 3.8.16
 

graph_datasets/utils/common.py

@@ -20,16 +20,18 @@ def format_result(
     timezone="Asia/Shanghai",
     **kwargs,
 ):
-    time = get_str_time(timezone)
     if sort_kw:
         kwargs = dict(sorted(kwargs.items()))
-    return {
-        **kwargs,
-        "ds": dataset,
-        "src": source,
-        "model": model,
-        "time": time,
-    }
+
+    kwargs.update(
+        {
+            "ds": dataset,
+            "src": source,
+            "model": model,
+            "time": get_str_time(timezone),
+        }
+    )
+    return kwargs
 
 
 def get_str_time(timezone="Asia/Shanghai"):
@@ -93,7 +95,8 @@ def tab_printer(
             [
                 k.replace("_", " "),
                 f"{args[k]}" if isinstance(args[k], bool) else format_value(args[k]),
-            ] for k in keys
+            ]
+            for k in keys
         ]
     )
     if cols_align is not None:
@@ -116,7 +119,9 @@ def download_tip(info: Dict) -> None:
         data_file (str): filepath.
         url (str): url for downloading.
     """
-    info["Tip"] = "If the download fails, \
+    info[
+        "Tip"
+    ] = "If the download fails, \
 use the 'Download URL' to download manually and move the file to the 'Save Path'."
 
     tab_printer(info)

graph_datasets/utils/evaluation/eval_tools.py

@@ -151,7 +151,7 @@ def kmeans_test(X, y, n_clusters, repeat=10):
     ari_list = []
     f1_list = []
     for _ in range(repeat):
-        kmeans = KMeans(n_clusters=n_clusters)
+        kmeans = KMeans(n_clusters=n_clusters, n_init='auto')
         y_pred = kmeans.fit_predict(X)
         (
             acc_score,