NDlib Integration for Network-Based Diffusion

This tutorial demonstrates how to use the NdlibInnovationModel to simulate diffusion on a network.

import networkx as nx
from innovate.abm.ndlib_model import NdlibInnovationModel

# Create a networkx graph
graph = nx.erdos_renyi_graph(n=100, p=0.1)

# Initialize the model
model = NdlibInnovationModel(num_agents=100, graph=graph)

# Run the simulation for 10 steps
for i in range(10):
    model.step()

# Get the status of the nodes
status = model.diffusion_model.status

# Count the number of adopted agents
adopted_agents = [agent for agent, status in status.items() if status == 'Infected']
print(f"Number of adopted agents: {len(adopted_agents)}")