s444409/Network_attack_propagation
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Refactor experiments

Browse Source
This commit is contained in:
Marcin Kostrzewski 2022-06-21 16:24:48 +02:00
parent 46ae4299c4
commit 8fe3c2bf8e
1 changed files with 20 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
network_attack_propagation.py
View File

@ -154,7 +154,7 @@ def degree_avg(edges, digits=2):
return round(mean(degrees.values()), digits) return round(mean(degrees.values()), digits)
def bus_network(n=30, infected_idx=0) -> tuple[Graph, float]: def bus_network(n=30, infected_idx=0) -> tuple[Graph, float, int]:
network = Graph() network = Graph()
nodes = [Node() for _ in range(n)] nodes = [Node() for _ in range(n)]
nodes[infected_idx].is_infected = True nodes[infected_idx].is_infected = True
@ -181,7 +181,7 @@ def star_network(cluster_count=5, starsize=6) -> tuple[Graph, float, int]:
return network, degree_avg(edges), node_count return network, degree_avg(edges), node_count
def ring_network(n=30) -> tuple[Graph, float]: def ring_network(n=30) -> tuple[Graph, float, int]:
network = Graph() network = Graph()
nodes = [Node() for _ in range(n)] nodes = [Node() for _ in range(n)]
nodes[0].is_infected = True nodes[0].is_infected = True
@ -200,51 +200,37 @@ def summary(average_degrees: list[float], propagation_speeds: list[float]) -> No
fig.savefig("summary.png") fig.savefig("summary.png")
plt.show() plt.show()
def bus_experiment():
def experiment(network, avg_degree, node_count):
degrees = [] degrees = []
speeds = [] speeds = []
sizes = [0, 8, 15] sizes = [0, 8, 15]
for i in sizes: for i in sizes:
bus, bus_avg_degree, node_count = bus_network(20 + i) do_graph_animation(f'bus{i}.gif', network, 90, nx.spiral_layout)
do_graph_animation(f'bus{i}.gif', bus, 90, nx.spiral_layout) speeds.append(network.rounds_survived / node_count)
speeds.append(bus.rounds_survived / node_count) degrees.append(avg_degree)
degrees.append(bus_avg_degree)
print(f"\n{node_count} NODE BUS") print(f"\n{node_count} NODE BUS")
print(f"average degree = {bus_avg_degree}") print(f"average degree = {avg_degree}")
print(f"propagation speed = {round(speeds[-1], 2)}") print(f"propagation speed = {round(speeds[-1], 2)}")
print(f"bus_{i} rounds survived = {bus.rounds_survived + 1}") print(f"bus_{i} rounds survived = {network.rounds_survived + 1}")
summary(degrees, speeds) summary(degrees, speeds)
def bus_experiment():
bus, bus_avg_degree, node_count = bus_network(20 + i)
experiment(bus, bus_avg_degree, node_count)
def ring_experiment(): def ring_experiment():
degrees = []
speeds = []
sizes = [0, 8, 15]
for i in sizes:
ring, ring_avg_degree, node_count = ring_network(20 + i) ring, ring_avg_degree, node_count = ring_network(20 + i)
do_graph_animation(f'ring{i}.gif', ring, 90, nx.spiral_layout) experiment(ring, ring_avg_degree, node_count)
speeds.append(ring.rounds_survived / node_count)
degrees.append(ring_avg_degree)
print(f"\n{node_count} NODE RING")
print(f"average degree = {ring_avg_degree}")
print(f"propagation speed = {round(speeds[-1], 2)}")
print(f"ring_{i} rounds survived = {ring.rounds_survived + 1}")
def star_experiment(): def star_experiment():
degrees = []
speeds = []
sizes = range(0, 8, 2)
for i in sizes:
star, star_avg_degree, node_count = star_network(cluster_count=3 + i, starsize=3 + i) star, star_avg_degree, node_count = star_network(cluster_count=3 + i, starsize=3 + i)
do_graph_animation(f'star{i}.gif', star, 120, nx.kamada_kawai_layout) experiment(star, star_avg_degree, node_count)
speeds.append(star.rounds_survived / node_count)
degrees.append(star_avg_degree)
print(f"\n{node_count} NODE STAR")
print(f"average degree = {star_avg_degree}")
print(f"propagation speed = {round(speeds[-1], 2)}")
print(f"star_{i} rounds survived = {star.rounds_survived + 1}")
summary(degrees, speeds)
def main(): def main():
bus_experiment() bus_experiment()