完整神经网络样例程序(TensorFlow,加入偏置项和relu激活函数)(源码)

这是一个使用TensorFlow构建的完整神经网络示例,包括了偏置项和ReLU激活函数的添加。程序通过随机生成的数据进行训练,展示了神经网络从初始化到训练的过程,并在训练过程中输出了关键参数和损失函数的变化。
摘要由CSDN通过智能技术生成
# coding=utf-8
"""完整的神经网络解决二分类问题(加入偏置项和relu激活函数)"""
import tensorflow as tf
from numpy.random import RandomState

# 定义训练数据batch大小
batch_size = 8

# 定义神经网络的参数
w1 = tf.Variable(tf.random_normal([2, 3], stddev=1, seed=1))
w2 = tf.Variable(tf.random_normal([3, 1], stddev=1, seed=1))
biases1 = tf.Variable(tf.constant(0.1,shape=[3]))
biases2 = tf.Variable(tf.constant(0.1,shape=[1]))


# 在shape的一个维度上使用None可以方便使用不大的batch大小,在训练时需要把数据分
# 成比较小的batch,但是在测试时,可以一次性使用全部的数据。当数据集比较小时这样比较
# 方便测试,但数据集比价大时,将大量数据放入一个batch可能会导致内存溢出。
x = tf.placeholder(tf.float32, shape=(None, 2), name='x-input')
y_ = tf.placeholder(tf.float32, shape=(None, 1), name='y-input')

# 定义神经网络前向传播的过程
a = tf.nn.relu(tf.
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值