extend ocr evaluation results

1 files changed, 7 insertions, 2 deletions

diff --git a/bjoern/videoanalyse/post_processing.py b/bjoern/videoanalyse/post_processing.py
index 445b72f..c295dc6 100644
--- a/bjoern/videoanalyse/post_processing.py
+++ b/bjoern/videoanalyse/post_processing.py
@@ -20,6 +20,7 @@ all_vp = [x for x in data_path.iterdir() if x.is_dir()]
 
 vp_results = []
 tab_results = []
+log_tab_results = []
 for vp_path in all_vp:
     log = LogParser.extract_activities(LogParser.get_log_data(vp_path))
     log = LogParser.generate_log(log)
@@ -45,11 +46,15 @@ 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_df = tab_switch.tab_switches_per_type(df, "group_index")
     tab_results.append(tab_df)
     tab_df.to_csv(f"{vp_path}/tabswitches.csv")
 
+    log_tab_df = tab_switch.tab_switches_per_type(df, "log_url")
+    log_tab_results.append(log_tab_df)
+    log_tab_df.to_csv(f"{vp_path}/log_tabswitches.csv")
+
     vp_results.append(df)
 
-evals = utils.evaluate_results(tab_results)
+evals = utils.evaluate_results(tab_results, log_tab_results)
 evals.to_csv(f"{data_path}/evaluation.csv")