fix names of semi supervised algorithms

alexgiving · Dec 1, 2024 · 09ed810 · 09ed810
1 parent 7ce2b91
commit 09ed810
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lkmeans/examples/main.py b/lkmeans/examples/main.py
@@ -18,8 +18,8 @@
 
 class ClusteringAlgorithmType(Enum):
     LKMEANS = 'lkmeans'
-    SOFT_SS_LKMEANS = 'soft_ss_lkmeans'
-    HARD_SS_LKMEANS = 'hard_ss_lkmeans'
+    SOFT_SEMI_SUPERVISED_LKMEANS = 'soft_semi_supervised_lkmeans'
+    HARD_SEMI_SUPERVISED_LKMEANS = 'hard_semi_supervised_lkmeans'
 
 
 class ExperimentArguments(Tap):
@@ -37,8 +37,8 @@ class ExperimentArguments(Tap):
 def get_clustering_algorithm(clustering_type: ClusteringAlgorithmType) -> Clustering:
     clustering_map: Dict[ClusteringAlgorithmType, Clustering] = {
         ClusteringAlgorithmType.LKMEANS: LKMeans,
-        ClusteringAlgorithmType.SOFT_SS_LKMEANS: SoftSemiSupervisedLKMeans,
-        ClusteringAlgorithmType.HARD_SS_LKMEANS: HardSemiSupervisedLKMeans
+        ClusteringAlgorithmType.SOFT_SEMI_SUPERVISED_LKMEANS: SoftSemiSupervisedLKMeans,
+        ClusteringAlgorithmType.HARD_SEMI_SUPERVISED_LKMEANS: HardSemiSupervisedLKMeans
     }
     return clustering_map[clustering_type]
 

diff --git a/lkmeans/examples/scripts/runner_supervised.sh b/lkmeans/examples/scripts/runner_supervised.sh
@@ -9,7 +9,7 @@ mkdir -p ${LOGDIR}
 MINKOSKI_VALUES=(0.5 1 2 5)
 T_VALUES=(0 0.2 0.4 0.6 0.8)
 N_POINTS_VALUES=(100 500 1000)
-CLUSTERINGS_VALUES=(soft_ss_lkmeans hard_ss_lkmeans)
+CLUSTERINGS_VALUES=(soft_semi_supervised_lkmeans hard_semi_supervised_lkmeans)
 SUPERVISION_RATIO_VALUES=(0.1 0.15 0.2)
 DIMENSION_VALUES=(20)
 NUM_CLUSTERS_VALUES=(2 3)