From 4d9c0dce1f5bc0bd3cde1b89875387f1a13c18c4 Mon Sep 17 00:00:00 2001 From: Niclas Dobbertin Date: Wed, 2 Aug 2023 14:13:45 +0200 Subject: add additional scripts --- bjoern/videoanalyse/eval.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 bjoern/videoanalyse/eval.py (limited to 'bjoern/videoanalyse/eval.py') diff --git a/bjoern/videoanalyse/eval.py b/bjoern/videoanalyse/eval.py new file mode 100644 index 0000000..a917b7a --- /dev/null +++ b/bjoern/videoanalyse/eval.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python3 + +import argparse +from pathlib import Path +import pandas as pd +import csv +from Levenshtein import distance as levendist + + +argparser = argparse.ArgumentParser( + description="Distance evaluation" +) +argparser.add_argument( + "vp_dir", help="Directory containing merged.csv" +) + +args = argparser.parse_args() + +vp_path = Path(args.vp_dir) + +df = pd.read_csv(vp_path / "merged.csv") + +df["levenshtein-distance"] = df.apply( + lambda row: levendist(str(row.url), str(row.log_url)), axis=1 +) + + + +df.to_csv(vp_path / "metrics.csv", quoting=csv.QUOTE_NONNUMERIC) -- cgit v1.2.3