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
Loading
......@@ -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