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