Python神经网络实验程序

最新推荐文章于 2024-07-25 23:52:06 发布

zhangzhao0907

最新推荐文章于 2024-07-25 23:52:06 发布

阅读量1.5k

点赞数 1

分类专栏： AI

本文链接：https://blog.csdn.net/zhangzhao0907/article/details/81836236

版权

该博客介绍了使用Python搭建的两层神经网络，隐藏层含有5个神经元，通过1000次梯度下降训练，应用于手写数字识别。

摘要由CSDN通过智能技术生成

代码来自《数据科学入门》。搭建了两层神经网络，隐藏层包含5个神经元，通过1000次梯度搜索训练模型参数，可以改进后识别手写数字。

from collections import Counter
from functools import partial
from linear_algebra import dot
import math, random
import matplotlib
import matplotlib.pyplot as plt

def step_function(x):
    return 1 if x >= 0 else 0

def perceptron_output(weights, bias, x):
    """returns 1 if the perceptron 'fires', 0 if not"""
    return step_function(dot(weights, x) + bias)

def sigmoid(t):
    return 1 / (1 + math.exp(-t))

def neuron_output(weights, inputs):
    return sigmoid(dot(weights, inputs))

def feed_forward(neural_network, input_vector):
    """takes in a neural network (represented as a list of lists of lists of weights)
    and returns the output from forward-propagating the input"""