从使用numpy array的方法创建numpy数组

Question

我有一个2D numpy数组作为基本sigmoid分类器的输入。 我希望分类器返回一个带有概率的数组

import numpy as np


def sigmoid(x):
    sigm = 1 / (1 + np.exp(-x))
    return sigm


def p(D, w, b):
    prob=sigmoid(np.dot(D[:][7],w)+b)
    return prob

如何更改p（），使其返回按输入数据顺序列出概率的1D numpy数组

Atm“prob”是一个长度为14的数组，但是输入数组“D”的大小超过400，因此在逻辑的某个地方有一个错误

Answer 1

编辑：问题是数组的切片不正确。必须使用D[:,7]而不是D[:][7]来提取列

---

可能是这样的：

import numpy as np


def sigmoid(x):
    sigm = 1 / (1 + np.exp(-x))
    return sigm


def p(D, w, b):
    prob=sigmoid(np.dot(D[:][7],w)+b)
    return np.ravel(prob)