pso神经网络用python_python - 如何通过PSO优化的神经网络代码使用不同的损失函数？ - 堆栈内存溢出...

我正在使用pyswarms PSO进行神经网络优化。 我正在尝试创建输入层和输出层的网络。

# Store the features as X and the labels as y

X = np.random.randn(25000,20)

y = np.random.random_integers(0,2,25000)

# In[29]:

def sigmoid(x):

return 1 / (1 + math.exp(-x))

# In[58]:

print(X_train.shape)

print(y_train.shape)

# In[63]:

# Forward propagation

def forward_prop(params):

"""Forward propagation as objective function

This computes for the forward propagation of the neural network, as

well as the loss. It receives a set of parameters that must be

rolled-back into the corresponding weights and biases.

Inputs

------

params: np.ndarray

The dimensions should include an unrolled version of the

weights and biases.

Returns

-------

float

The computed negative lo