process.py 文件源码

python
阅读 42 收藏 0 点赞 0 评论 0

项目:seqhawkes 作者: mlukasik 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号