From 5541099586dbd6018574cb44e1934907c121526f Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Thu, 3 Jul 2025 18:24:16 +0200 Subject: edit tool and add result analysis --- classification/tools/create_csv.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'classification/tools/create_csv.py') diff --git a/classification/tools/create_csv.py b/classification/tools/create_csv.py index c57c45e6a..ea997ccff 100755 --- a/classification/tools/create_csv.py +++ b/classification/tools/create_csv.py @@ -1,7 +1,13 @@ from os import listdir, path +from argparse import ArgumentParser root_directory = "../../results/classifier/" +parser = ArgumentParser() +parser.add_argument('-d', '--directory') + +args = parser.parse_args() + def parse_iteration(directory): dictionary = {} @@ -19,6 +25,11 @@ def output_csv(dictionary, full_path): file.write(f"{key}, {value}\n") def main(): + if args.directory: + dictionary = parse_iteration(args.directory) + output_csv(dictionary, args.directory) + exit() + for entry in listdir(root_directory): full_path = path.join(root_directory, entry) if path.isdir(full_path): -- cgit 1.4.1