手写数字识别tf1.0

import tensorflow as tf
mnist
mnist.train
mnist.validation
mnist.test
#图像显示函数
import matplotlib.pyplot as plt
def plot_image(image):
    plt.inshow(image.reshape(28,28),cmap="binary")
#输入数据
x=tf.placeholder(tf.float32,[None,784])
y=tf.placeholder(tf.float32,[None,10])
#定义变量
w=tf.Variable(tf.random_normal([784,10]))
b=tf.Variable(tf.zero([10]))
#模型构建
forward=tf.matmul(x,W)+b
pred=tf.nn.softmax(forward)
#损失函数
loss_function=#交叉熵函数
#优化函数
optimizer=tf.train.GradientDenscentOptimizer(learning_rate).minimize(loss_function)
#准确率
correct_prefiction=tf.equal(tf.argmax(pred,1),tf.argmax(y,1))
accuracy=tf.reduce_mean(tf.cast(correct_prediction,tf.float32)
#创建对话
sess=tf.Sessin()
init=tf.global_variables_initializer()
sess.run(init)
#训练模型
for epoch in range(train_epochs):
    for batch in range(total_batch):
        xs,ys=mnist.train.next_batc(batch_size)
        sess.run(optimizer,feed_dict={x:xs,y:ys})
    loss,acc = sess.run([loss_function,accuracy],feed_dict={x:mnist.validation.image,y:mnist.validation.;ables})
    if(epoch)%display_step==0:
        pass#打印准确率和损失值

#在测试集上评估模型准确率
accu_test = sess.run(accurary,feed_dict={x:mnist.test.images,y:mnist.test.lables})
#应用模型
prediction_result=sess.run(tf.argmax(pred,1),feed_dict={x:mnist.test.images})

多层神经网络

#两层隐藏层
H1_NN=256
H2_NN=64

w1=tf.Variable(tf.truncated_normal([784,H1_NN],stddev=0.1)
b1=tf.Variable(tf.zero([H1_NN))

w2=tf.Variable(tf.truncated_normal([H1_NN,H2_NN],stddev=0.1))
b2=tf.Variable(tf.zero([H2_NN]))

w3=tf.Variable(tf.truncated_normal([H2_NN,10],stddev=0.1))
b3=tf.Variable(tf.zero([10]))

Y1=tf.nn.relu(tf.matul(x,w1)+b1)
Y2=tf.nn.relu(tf.matul(x,w2)+b2)
forward=tf.matul(Y1,w3)+b3
pred=tf.nn.softmax(forward)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值