diff options
Diffstat (limited to 'diag/esan_color.py')
| -rw-r--r-- | diag/esan_color.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/diag/esan_color.py b/diag/esan_color.py index 7be050c..b283641 100644 --- a/diag/esan_color.py +++ b/diag/esan_color.py @@ -131,7 +131,8 @@ def main(): print(f"[esan/{args.grad_mode}] solve={best:.3f} LE AUROC={auc:.3f} mean_lam={lams.mean():+.3f} " f"passK={passk/n:.3f} lamsel={lamsel/n:.3f} ({time.time()-t0:.0f}s)") base = f"esan_{args.grad_mode}_none_n50_k3_p0.2_T3_ns3_s{args.seed}" - com = {'conv': 'esan', 'pe': 'none', 'grad_mode': args.grad_mode, 'contract': False, 'seed': args.seed} + com = {'conv': 'esan', 'pe': 'none', 'grad_mode': args.grad_mode, 'contract': False, + 'seed': args.seed, 'arch': 'rrog_once_agg_node_compute'} json.dump({**com, 'solve_rate': best}, open(os.path.join(OUT, f"color_{base}.json"), 'w')) json.dump({**com, 'auroc': float(auc), 'mean_lam': float(lams.mean())}, open(os.path.join(OUT, f"le_color_{base}.json"), 'w')) |