Add dataset to test and add hyperparameters to sample

sample/sample.cpp

@@ -1,31 +1,76 @@
 #include <iostream>
 #include <vector>
 #include <iomanip>
+#include <getopt.h>
 #include "../CPPFImdlp.h"
 #include "../tests/ArffFiles.h"
 
 using namespace std;
 using namespace mdlp;
 
+/* print a description of all supported options */
+void usage(const char* path)
+{
+    /* take only the last portion of the path */
+    const char* basename = strrchr(path, '/');
+    basename = basename ? basename + 1 : path;
 
-int main(int argc, char** argv)
+    cout << "usage: " << basename << "[OPTION]" << endl;
+    cout << "  -h, --help\t\t Print this help and exit." << endl;
+    cout << "  -f, --file[=FILENAME]\t {mfeat-factors, glass, iris, letter, kdd_JapaneseVowels, liver-disorders, test}." << endl;
+    cout << "  -m, --max_depth=INT\t max_depth pased to discretizer. Default = MAX_INT" << endl;
+    cout << "  -n, --min_length=INT\t interval min_length pased to discretizer. Default = 3" << endl;
+}
+
+tuple<string, int, int> parse_arguments(int argc, char** argv)
+{
+    string file_name;
+    int max_depth = numeric_limits<int>::max();
+    int min_length = 3;
+    static struct option long_options[] = {
+            { "help", no_argument, 0, 'h' },
+            { "file", required_argument, 0, 'f' },
+            { "max_depth", required_argument, 0, 'm' },
+            { "min_length", required_argument, 0, 'n' },
+            { 0, 0, 0, 0 }
+    };
+    while (1) {
+        auto c = getopt_long(argc, argv, "hf:m:n:", long_options, 0);
+        if (c == -1)
+            break;
+        switch (c) {
+            case 'h':
+                usage(argv[0]);
+                exit(0);
+            case 'f':
+                file_name = optarg;
+                break;
+            case 'm':
+                max_depth = atoi(optarg);
+                break;
+            case 'n':
+                min_length = atoi(optarg);
+                break;
+            case '?':
+                usage(argv[0]);
+                exit(1);
+            default:
+                abort();
+        }
+    }
+    if (file_name.empty()) {
+        usage(argv[0]);
+        exit(1);
+    }
+    return make_tuple(file_name, max_depth, min_length);
+}
+
+void process_file(string file_name, bool class_last, int max_depth, int min_length)
 {
     ArffFiles file;
     string path = "../../tests/datasets/";
-    map<string, bool> datasets = {
-            {"mfeat-factors",      true},
-            {"iris",               true},
-            {"letter",             true},
-            {"glass",              true},
-            {"kdd_JapaneseVowels", false},
-            {"test",               true}
-    };
-    if (argc != 2 || datasets.find(argv[1]) == datasets.end()) {
-        cout << "Usage: " << argv[0] << " {mfeat-factors, glass, iris, letter, kdd_JapaneseVowels, test}" << endl;
-        return 1;
-    }
 
-    file.load(path + argv[1] + ".arff", datasets[argv[1]]);
+    file.load(path + file_name + ".arff", class_last);
     auto attributes = file.getAttributes();
     int items = file.getSize();
     cout << "Number of lines: " << items << endl;
@@ -44,7 +89,7 @@ int main(int argc, char** argv)
         }
         cout << y[i] << endl;
     }
-    mdlp::CPPFImdlp test = mdlp::CPPFImdlp();
+    mdlp::CPPFImdlp test = mdlp::CPPFImdlp(min_length, max_depth);
     auto total = 0;
     for (auto i = 0; i < attributes.size(); i++) {
         auto min_max = minmax_element(X[i].begin(), X[i].end());
@@ -57,6 +102,33 @@ int main(int argc, char** argv)
         }
         total += test.getCutPoints().size();
     }
-    cout << "Total cut points: " << total << endl;
+    cout << "Total cut points ...: " << total << endl;
+    cout << "Total feature states: " << total + attributes.size() << endl;
+}
+
+
+int main(int argc, char** argv)
+{
+    map<string, bool> datasets = {
+            {"mfeat-factors",      true},
+            {"iris",               true},
+            {"letter",             true},
+            {"glass",              true},
+            {"kdd_JapaneseVowels", false},
+            {"liver-disorders",    true},
+            {"test",               true}
+    };
+    string file_name;
+    int max_depth, min_length;
+    tie(file_name, max_depth, min_length) = parse_arguments(argc, argv);
+    if (datasets.find(file_name) == datasets.end()) {
+        cout << "Invalid file name: " << file_name << endl;
+        usage(argv[0]);
+        exit(1);
+    }
+    process_file(file_name, datasets[file_name], max_depth, min_length);
+    cout << "File name: " << file_name << endl;
+    cout << "Max depth: " << max_depth << endl;
+    cout << "Min length: " << min_length << endl;
     return 0;
 }