Function alias_sample

ge/alias.py:41–54

:param accept: :param alias: :return: sample index

(accept, alias)

def alias_sample(accept, alias):
    """

    :param accept:
    :param alias:
    :return: sample index
    """
    N = len(accept)
    i = int(np.random.random() * N)
    r = np.random.random()
    if r < accept[i]:
        return i
    else:
        return alias[i]