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

鉴于:

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

  • 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)


梦里花落0921
浏览 110回答 1
1回答

慕田峪4524236

您尝试将shape数组分配给具有shapen的列m。更改以下out声明:out = np.zeros((m, n))或将预测分配给该行:for i, m in enumerate(m_models):    out[i, :] = m.predict(X)
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python