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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
import networkx as nx
import numpy as np
class InfluenceGraph(nx.Graph):
"""
networkX graph with mat and logmat attributes
"""
def __init__(self, max_proba, *args, **kwargs):
self.max_proba = max_proba
super(InfluenceGraph, self).__init__(*args, **kwargs)
def erdos_init(self, n, p):
G = nx.erdos_renyi_graph(n, p, directed=True)
self.add_nodes_from(G.nodes())
self.add_edges_from(G.edges())
@property
def mat(self):
if not hasattr(self, '_mat'):
self._mat = (self.max_proba * np.random.rand(len(self), len(self))
* np.asarray(nx.adjacency_matrix(self)))
return self._mat
@property
def logmat(self):
if not hasattr(self, '_logmat'):
self._logmat = np.log(1 - self.mat)
return self._logmat
def icc_cascade(G, p_init):
"""
Returns boolean vectors for one cascade
True means node was active at that time step
p_init: proba that node in seed set
"""
susceptible = np.ones(G.number_of_nodes(), dtype=bool)
active = np.random.rand(G.number_of_nodes()) < p_init
susceptible = susceptible - active
cascade = []
while active.any() and susceptible.any():
cascade.append(active)
active = np.exp(np.dot(G.logmat, active)) \
< np.random.rand(G.number_of_nodes())
active = active & susceptible
susceptible = susceptible - active
return cascade
def concat_cascades(cascades):
"""
Concatenate list of cascades into matrix
"""
def test():
"""
unit test
"""
G = InfluenceGraph(max_proba = .3)
G.erdos_init(n = 100, p = 1)
import time
t0 = time.time()
print len(icc_cascade(G, p_init=.1))
t1 = time.time()
print t1 - t0
if __name__ == "__main__":
test()
|
