X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=clueless-kmean.cc;h=557f0d8d2a161fd789aac9d2d2d28c57fe20629d;hb=2ac817885f81e54fa67ad42a2fb4f46304085089;hp=13d96242f3fb067d952ddccf51ab504d004b404b;hpb=ca6d045155d4c948063f49b6de8c35c0e3246e7a;p=clueless-kmeans.git

diff --git a/clueless-kmean.cc b/clueless-kmean.cc
index 13d9624..557f0d8 100644
--- a/clueless-kmean.cc
+++ b/clueless-kmean.cc
@@ -76,7 +76,23 @@ int main(int argc, char **argv) {
 
   glp_term_out(0);
 
-  clusterer.train(Clusterer::UNINFORMATIVE_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,