def spontaneousnode_count(
infecting_vec,
infected_vec,
node_vec,
D,
):
'''
Returns a vector with count of spontanous infections for nodes.
Arguments:
infecting_vec - vector of infecting event ids
infected_vec - vector of event ids
node_vec - vector of infected node ids
D - number of nodes
'''
spontaneous_nodes = node_vec[infecting_vec == infected_vec]
updates = np.zeros((D, 1))
for node in spontaneous_nodes:
updates[int(node)] += 1
return updates
评论列表
文章目录