python实现N-HiTS代码

以下是一个基本的 Python 实现 N-HiTS 的代码:

import numpy as np

def n_hits(n, p):
    """
    :param n: int, the number of trials
    :param p: float, the probability of success
    :return: int, the number of successes in n trials
    """
    successes = 0
    for i in range(n):
        if np.random.uniform() < p:
            successes += 1
    return successes

这个函数使用 NumPy 库中的随机数生成器来模拟 Bernoulli 实验。在每次实验中，函数生成一个在0和1之间均匀分布的随机数，如果这个随机数小于成功概率 `p`，则将成功次数增加1。最后，函数返回成功次数。