python实现N-HiTS代码
时间: 2023-10-01 16:06:26 浏览: 104
实现一个数的N次方源码
以下是一个基本的 Python 实现 N-HiTS 的代码:
```python
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。最后,函数返回成功次数。
阅读全文