AI写算法:BP神经网络

当编写一个基于Python的BP(反向传播)神经网络的代码时,通常我们会使用像NumPy这样的库来处理矩阵运算,或者使用更高级的库,如TensorFlowPyTorch。以下是一个简单的使用NumPy实现的BP神经网络的例子。

请注意,这个例子是为了教学目的而简化的,并没有包含所有可能的优化和特性(如学习率衰减、动量、批量归一化、正则化等)。


python复制代码

import numpy as np
# Sigmoid激活函数
def sigmoid(x):
return 1 / (1 + np.exp(-x))
# Sigmoid的导数
def sigmoid_derivative(x):
return x * (1.0 - x)
class NeuralNetwork:
def __init__(self, x, y):
self.input = x
self.weights1 = np.random.rand(self.input.shape[1],4)
self.weights2 = np.random.rand(4,1)
self.y = y
self.output = np.zeros(self.y.shape)
def feedforward(self):
self.layer1 = sigmoid(np.dot(self.input, self.weights1))
self.output = sigmoid(np.dot(self.layer1, self.weights2))
def backprop(self):
# 反向传播算法
# 应用链式法则计算权重的梯度
d_weights2 = np.dot(self.layer1.T, (2*(self.y - self.output) * sigmoid_derivative(self.output)))
d_weights1 = np.dot(self.input.T, (np.dot(2*(self.y - self.output) * sigmoid_derivative(self.output), self.weights2.T) * sigmoid_derivative(self.layer1)))
# 更新权重
self.weights1 += d_weights1
self.weights2 += d_weights2
def train(self, epochs):
for i in range(epochs):
self.feedforward()
self.backprop()
# 示例数据
X = np.array([[0,0,1],
[0,1,1],
[1,0,1],
[1,1,1]])
y = np.array([[0],[1],[1],[0]])
nn = NeuralNetwork(X,y)
nn.train(1500)
print(nn.output)

这个神经网络有两个隐藏层神经元,一个输出层神经元,用于处理简单的XOR问题。请注意,在实际应用中,可能需要调整网络结构(如添加更多层、更改神经元数量等)以及训练参数(如学习率、迭代次数等)以达到最佳性能。同时,对于更复杂的问题,使用TensorFlowPyTorch等库通常会更加高效和方便。

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

军哥说AI

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值