Skip to content
Snippets Groups Projects
Commit c2b03d9b authored by Fize Jacques's avatar Fize Jacques
Browse files

Debug

parent f5f22683
No related branches found
No related tags found
No related merge requests found
......@@ -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))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment