diff options
Diffstat (limited to 'bjoern/videoanalyse/post_processing.py')
-rw-r--r-- | bjoern/videoanalyse/post_processing.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/bjoern/videoanalyse/post_processing.py b/bjoern/videoanalyse/post_processing.py index b5e3c65..445b72f 100644 --- a/bjoern/videoanalyse/post_processing.py +++ b/bjoern/videoanalyse/post_processing.py @@ -8,6 +8,7 @@ import pandas as pd import LogParser import utils +import tab_switch argparser = argparse.ArgumentParser(description="OCR-Logfile evaluation") argparser.add_argument("vp_dir", help="Directory with all VPs") @@ -18,6 +19,7 @@ data_path = Path(args.vp_dir) all_vp = [x for x in data_path.iterdir() if x.is_dir()] vp_results = [] +tab_results = [] for vp_path in all_vp: log = LogParser.extract_activities(LogParser.get_log_data(vp_path)) log = LogParser.generate_log(log) @@ -43,8 +45,11 @@ for vp_path in all_vp: df = pd.read_csv(f"{vp_path}/metrics_grps.csv") + tab_df = tab_switch.tab_switches_per_type(df) + tab_results.append(tab_df) + tab_df.to_csv(f"{vp_path}/tabswitches.csv") + vp_results.append(df) -evals = utils.evaluate_results(vp_results) -pprint(evals) +evals = utils.evaluate_results(tab_results) evals.to_csv(f"{data_path}/evaluation.csv") |