X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=clueless-kmean.cc;h=557f0d8d2a161fd789aac9d2d2d28c57fe20629d;hb=056eef1e23b3f6e5218b7bc3800f6412a3f97bfc;hp=1eea96fa91b6b882e04e72098dae07aa7ab8dd44;hpb=2455f83ba251602d5e04640067094f09f03aaa3d;p=clueless-kmeans.git

diff --git a/clueless-kmean.cc b/clueless-kmean.cc
index 1eea96f..557f0d8 100644
--- a/clueless-kmean.cc
+++ b/clueless-kmean.cc
@@ -76,7 +76,25 @@ int main(int argc, char **argv) {
 
   glp_term_out(0);
 
-  clusterer.train(nb_clusters, sample_set.dim,
+  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,
                   sample_set.nb_classes, sample_set.labels,
                   associated_clusters);
@@ -107,4 +125,6 @@ int main(int argc, char **argv) {
   }
 
   delete[] associated_clusters;
+
+  glp_free_env(); // I do not want valgrind to complain
 }