diff --git a/run_eval.py b/run_eval.py index 6567a89735e0f7166a98c8674b85c7933d38f83a..985f55c605217c3904f2031b90d567f587370c34 100644 --- a/run_eval.py +++ b/run_eval.py @@ -15,6 +15,7 @@ parser = argparse.ArgumentParser() parser.add_argument("dataset_dir") parser.add_argument("output_filename") parser.add_argument("-f", "--format", default="gexf", choices=["gexf", "gml", "txt"]) +parser.add_argument("-t","--train-frac",default=0.9,type=float) args = parser.parse_args() fns = sorted(glob.glob(args.dataset_dir + "/*." + args.format)) @@ -25,7 +26,7 @@ for fn in pbar: pbar.set_description("run eval on "+ fn) if not os.path.exists(fn + "_results_lp"): - command = "python evalNE_script.py {0} -f {1} -n".format(fn, args.format).split() + command = "python evalNE_script.py {0} -f {1} -n --train-frac {2}".format(fn, args.format,args.train_frac).split() output = subprocess.run(command) if not output.returncode == 0: print("Error! for the command :", " ".join(command))