Independent Cascade

def get_ndlib_model(graph, seed_set):
    model = ep.IndependentCascadesModel(g)
    config = mc.Configuration()
    config.add_model_initial_configuration("Infected", seed_set)
    for e in graph.edges():
        config.add_edge_configuration("threshold", e, random.uniform(0,1))
    model.set_initial_status(config)
    return model

g = nx.erdos_renyi_graph(1000, 0.1)
S = [] # some seed set
for i in range(5):
    S.append(int(random.uniform(1, 999)))

model = get_ndlib_model(g, S)
iterations = model.iteration_bunch(len(g.edges))