如何在python中生成异方差数据

Question

我正在尝试使用NumPy生成Heteroscedasticity数据。代码如下所示

def generate_data(
    size: int, bounds: Tuple[int, int] = (0, 1000)
) -> Tuple[ndarray, ndarray, ndarray]:
    X = np.random.uniform(bounds[0], bounds[1], size=size)
    sd_deviation = np.random.uniform(0, 4, size=size) + (X / 13)
    error = np.random.normal(0, sd_deviation, size=size)
    y_train = error + X
    plt.plot(y_train, "o")
    plt.show()

但是我没有从图中得到清晰的Heteroscedasticity输出

你能告诉我如何生成异方差数据吗

Answer 1

只需更改X = np.random.uniform(bounds[0], bounds[1], size=size)

与X = np.arange(bounds[0], size)

您正在生成uniform分布数据（我不确定您是否可以使用统一分布来实现这一点）