rmontanana/mdlp
1
1
Fork 0
mirror of https://github.com/rmontanana/mdlp.git synced 2025-08-15 23:45:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add path argument to command line

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-02-28 10:52:26 +01:00
parent 200015000c
commit c00b7a613c
1 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
sample/sample.cpp
View File

@@ -21,24 +21,27 @@ void usage(const char* path)
cout << "usage: " << basename << "[OPTION]" << endl;
cout << " -h, --help\t\t Print this help and exit." << endl;
cout << " -f, --file[=FILENAME]\t {all, glass, iris, kdd_JapaneseVowels, letter, liver-disorders, mfeat-factors, test}." << endl;
cout << " -p, --path[=FILENAME]\t folder where the arff dataset is located, default " << PATH << 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)
tuple<string, string, int, int> parse_arguments(int argc, char** argv)
{
string file_name;
string path = PATH;
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' },
{ "path", required_argument, 0, 'p' },
{ "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);
auto c = getopt_long(argc, argv, "hf:p:m:n:", long_options, 0);
if (c == -1)
break;
switch (c) {
@@ -54,6 +57,11 @@ tuple<string, int, int> parse_arguments(int argc, char** argv)
case 'n':
min_length = atoi(optarg);
break;
case 'p':
path = optarg;
if (path.back() != '/')
path += '/';
break;
case '?':
usage(argv[0]);
exit(1);
@@ -65,14 +73,14 @@ tuple<string, int, int> parse_arguments(int argc, char** argv)
usage(argv[0]);
exit(1);
}
return make_tuple(file_name, max_depth, min_length);
return make_tuple(file_name, path, max_depth, min_length);
}
void process_file(string file_name, bool class_last, int max_depth, int min_length)
void process_file(string path, string file_name, bool class_last, int max_depth, int min_length)
{
ArffFiles file;
file.load(PATH + file_name + ".arff", class_last);
file.load(path + file_name + ".arff", class_last);
auto attributes = file.getAttributes();
int items = file.getSize();
cout << "Number of lines: " << items << endl;
@@ -108,14 +116,14 @@ void process_file(string file_name, bool class_last, int max_depth, int min_leng
cout << "Total feature states: " << total + attributes.size() << endl;
}
void process_all_files(map<string, bool> datasets, int max_depth, int min_length)
void process_all_files(map<string, bool> datasets, string path, int max_depth, int min_length)
{
cout << "Results: " << "Max_depth: " << max_depth << " Min_length: " << min_length << endl << endl;
printf("%-20s %4s %4s\n", "Dataset", "Feat", "Cuts Time(s)");
printf("==================== ==== ==== =======\n");
for (auto dataset : datasets) {
ArffFiles file;
file.load(PATH + dataset.first + ".arff", dataset.second);
file.load(path + dataset.first + ".arff", dataset.second);
auto attributes = file.getAttributes();
vector<samples_t>& X = file.getX();
labels_t& y = file.getY();
@@ -145,18 +153,18 @@ int main(int argc, char** argv)
{"mfeat-factors", true},
{"test", true}
};
string file_name;
string file_name, path;
int max_depth, min_length;
tie(file_name, max_depth, min_length) = parse_arguments(argc, argv);
tie(file_name, path, max_depth, min_length) = parse_arguments(argc, argv);
if (datasets.find(file_name) == datasets.end() && file_name != "all") {
cout << "Invalid file name: " << file_name << endl;
usage(argv[0]);
exit(1);
}
if (file_name == "all")
process_all_files(datasets, max_depth, min_length);
process_all_files(datasets, path, max_depth, min_length);
else {
process_file(file_name, datasets[file_name], max_depth, min_length);
process_file(path, 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;