一次应用一列后的numpy数组的平均值

Question

给出：

• m个模型，每个模型都有一个predict函数，该函数采用一个形状为[n_samples，n_features]的numpy数组，并返回一个形状为[n_samples]的数组，如果模型表示yes，则包含1否则为0。
• X个形状为[n_samples，n_features]的数据点

预期的输出：对于每个数据点，发出了多少个模型。

out = np.zeros((n, m))
for i, m in enumerate(m_models):
    out[:, i] = m.predict(X)    # Doesn't seem to work
scores = np.mean(out, axis=1)

Answer 1

您尝试将形状为n的数组分配给形状为m的列。在以下位置更改您的out声明：

out = np.zeros((m, n))

或将预测分配给该行：

for i, m in enumerate(m_models):
    out[i, :] = m.predict(X)