Network-Immunization/run_solver.py at master · allogn/Network-Immunization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import os
import sys
import time
import argparse
import numpy as np
import json
import pickle as pkl
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
from DegreeSolver import *
from DomSolver import *
from RandomSolver import *
from NetShapeSolver import *
from NetShieldSolver import *
from Simulator import *

if __name__ == "__main__":
    t1 = time.time()
    parser = argparse.ArgumentParser(description="Run solver on a single graph with seeds")
    parser.add_argument("graph", type=str)
    parser.add_argument("seeds", type=str)
    parser.add_argument("nodes_to_block", type=int)
    parser.add_argument("algorithm", type=str)
    parser.add_argument("-j", "--simulation_iterations", type=int, default="100")
    parser.add_argument("-o", "--outfile", type=str, default="a.out")
    parser.add_argument("-p", "--other_params", type=str, nargs="*")
    args = parser.parse_args()

    G, seeds = pkl.load(open(args.graph,'rb')), np.atleast_1d(np.loadtxt(args.seeds))
    k = args.nodes_to_block
    problem_params = {}
    if args.other_params:
        for i in range(int(len(args.other_params)/2)):
            if args.problem_params[2*i+1].isdigit():
                problem_params[args.other_params[2*i]] = int(args.other_params[2*i+1])
            else:
                problem_params[args.other_params[2*i]] = args.other_params[2*i+1]
    z = dict(problem_params)

    Solver = eval(args.algorithm + "Solver")
    solver = Solver(G, seeds, k, **z)
    solver.run()
    print("%s blocked %d nodes in a graph of size %d." % (solver.get_name(), k, len(G)))
    print("Running simulations...")

    simulator = Simulator(G, seeds)
    simulator.add_blocked(0, solver.log['Blocked nodes'])
    results = simulator.run(args.simulation_iterations)
    solver.log.update({"simulation": results['solvers'][0]})
    json.dump(solver.log, open(args.outfile, "w"))
    print("Solver Time: %1.5fs; Objective (saved): %1.1f; Total time: %1.5s" % (solver.log["Total time"], results['solvers'][0]["saved nodes"]["mean"], (time.time() - t1)))
    print("Logs saved to {}.".format(args.outfile))