From f8b664ba25c1b6d42fac073ff73a7485c5cecb92 Mon Sep 17 00:00:00 2001
From: Fize Jacques <jacques.fize@cirad.fr>
Date: Tue, 12 Jan 2021 23:09:18 +0100
Subject: [PATCH] add run_eval script
---
run_eval.py | 35 +++++++++++++++++++++++++++++++++++
1 file changed, 35 insertions(+)
create mode 100644 run_eval.py
diff --git a/run_eval.py b/run_eval.py
new file mode 100644
index 0000000..352e7ba
--- /dev/null
+++ b/run_eval.py
@@ -0,0 +1,35 @@
+# coding = utf-8
+
+import glob
+import subprocess
+from lib.helpers import parse_evalne_output
+from lib.utils import load_edgelist
+import os
+import pandas as pd
+from tqdm import tqdm
+
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("dataset_dir")
+parser.add_argument("output_filename")
+
+args = parser.parse_args()
+fns = glob.glob(args.dataset_dir + "/*.txt")
+
+all_res = []
+for fn in tqdm(fns):
+ print("run eval on ", fn)
+ command = "python evalNE_script.py {0} -v".format(fn).split()
+ output = subprocess.run(command)
+ if not output.returncode == 0:
+ print("Error! for the command :", " ".join(command))
+ continue
+ df_results = parse_evalne_output(open(fn + "_results_lp").read())
+ name = os.path.basename(fn)
+ edge_len = len(pd.read_csv(fn, sep="\t", header=None))
+ df_results["nb_edge"] = edge_len
+ df_results["filename"] = name
+ all_res.append(df_results)
+
+pd.concat(all_res).to_csv(args.output_filename,sep="\t",index=None)
\ No newline at end of file
--
GitLab