1 files changed, 11 insertions, 0 deletions
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):