np.sum 行轴在 Numpy 中不起作用

Question

我写了一个 softmax 回归函数 def softmax_1(x)，它本质上接受一个 m x n 矩阵，对矩阵取幂，然后对每一列的指数求和。

x = np.arange(-2.0, 6.0, 0.1)
scores = np.vstack([x, np.ones_like(x), 0.2 * np.ones_like(x)])
#scores shape is (3, 80)

def softmax_1(x):
    """Compute softmax values for each sets of scores in x."""
    return(np.exp(x)/np.sum(np.exp(x),axis=0))

将其转换为 DataFrame 我必须转置

DF_activation_1 = pd.DataFrame(softmax_1(scores).T,index=x,columns=["x","1.0","0.2"])

所以我想尝试制作一个接受转置版本并计算 softmax 函数的 softmax 函数

scores_T = scores.T
#scores_T shape is (80,3)

def softmax_2(y):
    return(np.exp(y/np.sum(np.exp(y),axis=1)))

DF_activation_2 = pd.DataFrame(softmax_2(scores_T),index=x,columns=["x","1.0","0.2"])

然后我得到这个错误：

Traceback (most recent call last):
  File "softmax.py", line 22, in <module>
    DF_activation_2 = pd.DataFrame(softmax_2(scores_T),index=x,columns=["x","1.0","0.2"])
  File "softmax.py", line 18, in softmax_2
    return(np.exp(y/np.sum(np.exp(y),axis=1)))
ValueError: operands could not be broadcast together with shapes (80,3) (80,)

为什么我在 np.sum 方法中转置和切换轴时不起作用？

Answer 1

改变

np.exp(y/np.sum(np.exp(y),axis=1))

到

np.exp(y)/np.sum(np.exp(y),axis=1, keepdims=True)

这意味着 np.sum 将 return 形状为 (80, 1) 而不是 (80,) 的数组，这将为除法正确广播。另请注意括号关闭的更正。

np.sum 行轴在 Numpy 中不起作用

np.sum for row axis not working in Numpy

python

numpy

machine-learning

matrix

softmax