X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=clueless-kmeans.git;a=blobdiff_plain;f=clueless-kmean.cc;fp=clueless-kmean.cc;h=557f0d8d2a161fd789aac9d2d2d28c57fe20629d;hp=e3d4f93a7498c9bfd5430a0561aeb5d14f079edd;hb=379b91965f8d18b05cc40c4a14044eda29a6e82c;hpb=1a7c12097efbe2cb377f4ef39e1564ba1ac42f4b

diff --git a/clueless-kmean.cc b/clueless-kmean.cc
index e3d4f93..557f0d8 100644
--- a/clueless-kmean.cc
+++ b/clueless-kmean.cc
@@ -76,8 +76,23 @@ int main(int argc, char **argv) {
 
   glp_term_out(0);
 
-  clusterer.train(Clusterer::UNINFORMATIVE_LP_ASSOCIATION,
-                  // Clusterer::STANDARD_LP_ASSOCIATION,
+  int mode;
+
+  if(argc == 2) {
+    if(strcmp(argv[1], "standard") == 0) {
+      mode = Clusterer::STANDARD_LP_ASSOCIATION;
+    } else if(strcmp(argv[1], "clueless") == 0) {
+      mode = Clusterer::UNINFORMATIVE_LP_ASSOCIATION;
+    } else {
+      cerr << "Unknown association mode " << argv[1] << endl;
+      exit(EXIT_FAILURE);
+    }
+  } else {
+    cerr << "Usage: " << argv[0] << " standard|clueless" << endl;
+    exit(EXIT_FAILURE);
+  }
+
+  clusterer.train(mode,
                   nb_clusters,
                   sample_set.dim,
                   sample_set.nb_points, sample_set.points,