Add diabetes dataset to sample

This commit is contained in:
2023-06-12 12:34:04 +02:00
parent b21e85f5e8
commit e8559faf1f
3 changed files with 874 additions and 1 deletions

View File

@@ -23,7 +23,7 @@ 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}."
<< " -f, --file[=FILENAME]\t {all, diabetes, 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;
@@ -164,6 +164,7 @@ void process_all_files(const map<string, bool>& datasets, const string& path, in
int main(int argc, char** argv)
{
map<string, bool> datasets = {
{"diabetes", true},
{"glass", true},
{"iris", true},
{"kdd_JapaneseVowels", false},