Added a new "non-absolute" criterion that enforces the proportions of classes in...

[clueless-kmeans.git] / clueless-kmeans.cc

diff --git a/clueless-kmeans.cc b/clueless-kmeans.cc
index 14e5b92..d71a3cf 100644 (file)
--- a/clueless-kmeans.cc
+++ b/clueless-kmeans.cc
@@ -83,12 +83,14 @@ int main(int argc, char **argv) {
       mode = Clusterer::STANDARD_LP_ASSOCIATION;
     } else if(strcmp(argv[1], "clueless") == 0) {
       mode = Clusterer::UNINFORMATIVE_LP_ASSOCIATION;
+    } else if(strcmp(argv[1], "clueless-absolute") == 0) {
+      mode = Clusterer::UNINFORMATIVE_LP_ASSOCIATION_ABSOLUTE;
     } else {
       cerr << "Unknown association mode " << argv[1] << endl;
       exit(EXIT_FAILURE);
     }
   } else {
-    cerr << "Usage: " << argv[0] << " standard|clueless" << endl;
+    cerr << "Usage: " << argv[0] << " standard|clueless|clueless-absolute" << endl;
     exit(EXIT_FAILURE);
   }