在 numpy 中实现零均值和单位方差

Question

我得到了一个函数的定义，并要求我按如下方式实现它：

# Problem 1 - Apply zero mean and zero variance scale to the image features
def normalize(data):
    pass

然后使用numpy 提供单元测试，这将断言我的实现成功

编辑

这不是我的单元测试，而是由课程讲师布置的。

np.testing.assert_array_almost_equal(
    normalize(np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])),
    np.array([-0.4, -0.3, -0.2, -0.099, 0.0, 0.099, 0.199, 0.3, 0.4, 0.5]),
    decimal=3)

我的解决办法是

def normalize(data):
    return ((data - data.mean()) / data.max() - data.min())

但我的理解肯定有差距，因为我正在理解

AssertionError: 
Arrays are not almost equal to 3 decimals

(mismatch 100.0%)
 x: array([-1.45, -1.35, -1.25, -1.15, -1.05, -0.95, -0.85, -0.75, -0.65, -0.55])
 y: array([-0.4  , -0.3  , -0.2  , -0.099,  0.   ,  0.099,  0.199,  0.3  ,
        0.4  ,  0.5  ])

通过谷歌搜索我也找到了

(data - data.mean()) / data.std()

经过更多搜索，我尝试了

(data - data.mean()) / data.var()

但两种解决方案都没有正确断言。

那么这里的正确实现是什么？

Answer 1

首先你可能打算这样做：

(data - data.mean()) / (data.max() - data.min())

而不是这个：

((data - data.mean()) / data.max() - data.min())

那么这不是规范化的正确定义。相反，正如您通过搜索发现的那样，正确的定义是：

(data - data.mean()) / data.std()

最后你的单元测试是错误的，所以当使用正确的定义调用它时它会失败。

Answer 2

我认为您的问题可能有错字。请注意，如果您使用公式，则单元测试通过

(data - data.mean()) / data.var()

但前提是您将 0 附加到测试输入数组的开头，并将 -0.5 附加到输出数组的开头。如果不是笔误，那就是奇怪的巧合。