process.py 文件源码

def spontaneousmeme_count(
    infecting_vec,
    infected_vec,
    eventmemes,
    M,
    ):
    '''
    Returns a vector with count of spontanous infections for memes.

    Arguments:
    infecting_vec - vector of infecting event ids
    infected_vec - vector of event ids
    eventmemes - vector of meme ids
    M - number of memes
    '''

    spontaneous_memes = eventmemes[infecting_vec == infected_vec]
    updates = np.zeros((M, 1))
    for meme in spontaneous_memes:
        updates[int(meme)] += 1
    return updates