python实现随机梯度下降(SGD)

最新推荐文章于 2024-07-31 14:47:45 发布

芳草碧连天lc

最新推荐文章于 2024-07-31 14:47:45 发布

阅读量9.7k

点赞数 3

分类专栏：机器学习文章标签： python 神经网络

本文链接：https://blog.csdn.net/leichaoaizhaojie/article/details/56840328

版权

使用神经网络进行样本训练，要实现随机梯度下降算法。这里我根据麦子学院彭亮老师的讲解，总结如下，（神经网络的结构在另一篇博客中已经定义）：

def SGD(self, training_data, epochs, mini_batch_size, eta, test_data=None):
    if test_data:
        n_test = len(test_data)#有多少个测试集
        n = len(training_data)
        for j in xrange(epochs):
            random.shuffle(training_data)
            mini_batches = [
                training_data[k:k+mini_batch_size] 
                for k in xrange(0,n,mini_batch_size)]
            for mini