In [1]:
addprocs(7)
Out[1]:
In [2]:
@everywhere begin
using NetworkDiscovery
using POMDPs
using POMDPToolbox
end
In [3]:
rng=MersenneTwister(1)
nodes = 1000
comms = 5
probes = 30
p_inter = 0.01
p_intra = 0.3
N = 1000
Out[3]:
In [4]:
function est_rew(policy, nodes, comms, probes, p_intra, p_inter, N)
sum = @parallel (+) for i in 1:N
prob_rng = MersenneTwister(i)
sim_rng = MersenneTwister(i)
nw = generate_network(rng, nodes, comms, p_intra, p_inter)
pomdp = generate_problem(rng, nw, probes, 1, 100.0, 10, 10, p_intra, p_inter)
revealed = initial_belief(pomdp)
sim = RolloutSimulator(rng=sim_rng, initial_state=nw, initial_belief=revealed)
simulate(sim, pomdp, policy)
end
end
Out[4]:
In [5]:
policy = DiscoveryHeuristic(ProbeHighestDegree(true), GuessBasedOnNeighbors(rng))
Out[5]:
In [6]:
@time est_rew(policy, nodes, comms, probes, p_intra, p_inter, N)
Out[6]:
In [ ]: