给我个ANN二分类的代码
时间: 2024-05-08 11:18:07 浏览: 14
当然可以,下面是一个简单的ANN二分类的代码:
```
import numpy as np
# 定义sigmoid函数
def sigmoid(x):
return 1 / (1 + np.exp(-x))
# 定义神经网络类
class NeuralNetwork:
def __init__(self, input_size, hidden_size, output_size):
# 初始化权重矩阵
self.weights1 = np.random.rand(input_size, hidden_size)
self.weights2 = np.random.rand(hidden_size, output_size)
# 前向传播
def forward(self, inputs):
# 计算隐藏层输出
hidden = sigmoid(np.dot(inputs, self.weights1))
# 计算输出层输出
output = sigmoid(np.dot(hidden, self.weights2))
return output
# 测试
inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]])
targets = np.array([[0], [1], [1], [0]])
nn = NeuralNetwork(2, 4, 1)
for i in range(10000):
# 前向传播
output = nn.forward(inputs)
# 计算误差
error = targets - output
# 反向传播
output_delta = error * output * (1 - output)
hidden_delta = np.dot(output_delta, nn.weights2.T) * hidden * (1 - hidden)
# 更新权重矩阵
nn.weights2 += np.dot(hidden.T, output_delta)
nn.weights1 += np.dot(inputs.T, hidden_delta)
print(nn.forward(inputs))
```
希望能对你有所帮助!