多层感知机(multi-layer perceptron)实现手写体分类(TensorFlow)
来源:互联网 发布:西门淘宝店网址 编辑:程序博客网 时间:2024/05/19 12:39
#multi_layer Perceptron by ffzhangimport numpy as npimport tensorflow as tffrom tensorflow.examples.tutorials.mnist import input_dataimport timeimport osos.environ["CUDA_VISIBLE_DEVICES"]='2'mnist = input_data.read_data_sets('data/mnist',one_hot=True)mnist.train.images.shapemnist.train.labels.shape#define placeholder to save the training dataX = tf.placeholder(tf.float32,[None,784],name='X_placeholder')Y = tf.placeholder(tf.float32,[None,10],name='Y_placeholder')#define parametersn_hidden_1 =256n_hidden_2 =256n_input =784n_classes =10weights ={ 'h1': tf.Variable(tf.random_normal([n_input,n_hidden_1]),name='W1'), 'h2': tf.Variable(tf.random_normal([n_hidden_1,n_hidden_2]),name='W2'), 'out': tf.Variable(tf.random_normal([n_hidden_2,n_classes]),name='W')}biases ={ 'b1': tf.Variable(tf.random_normal([n_hidden_1]),name='b1'), 'b2': tf.Variable(tf.random_normal([n_hidden_2]),name='b2'), 'out': tf.Variable(tf.random_normal([n_classes]),name='bias')}#define graph for this networkdef multilayer_perceptron(x,weights,biases): layer_1 = tf.add(tf.matmul(x,weights['h1']),biases['b1'],name='fc_1') layer_1 = tf.nn.relu(layer_1,name='relu_1') layer_2 = tf.add(tf.matmul(layer_1,weights['h2']),biases['b2'],name='fc_2') layer_2 = tf.nn.relu(layer_2,name='relu_2') out_layer = tf.add(tf.matmul(layer_2,weights['out']),biases['out'],name='fc_3') return out_layerpred = multilayer_perceptron(X, weights, biases)learning_rate = 0.001loss_all = tf.nn.softmax_cross_entropy_with_logits(labels=Y,logits=pred,name='cross_entropy')loss = tf.reduce_mean(loss_all,name='avg_loss')optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(loss)init =tf.global_variables_initializer()training_epochs=15batch_size=128display_step =1with tf.Session() as sess: sess.run(init) writer =tf.summary.FileWriter('./graphs/MLP_dnn',sess.graph) for epoch in range(training_epochs): avg_loss =0 total_batch = int(mnist.train.num_examples/batch_size) for i in range(total_batch): batch_x, batch_y =mnist.train.next_batch(batch_size) _, l =sess.run([optimizer,loss],feed_dict={X: batch_x, Y: batch_y}) avg_loss += l/total_batch if epoch%display_step==0: print ('Epoch:', '%04d'% (epoch+1),'cost=', "{:.9f}".format(avg_loss)) print ('optimizer finished') correct_prediction = tf.equal(tf.argmax(pred,1),tf.argmax(Y,1)) accuracy = tf.reduce_mean(tf.cast(correct_prediction,'float')) print ('accuracy:',accuracy.eval({X:mnist.test.images, Y:mnist.test.labels})) writer.close()
阅读全文
0 0
- 多层感知机(multi-layer perceptron)实现手写体分类(TensorFlow)
- Theano Multi Layer Perceptron 多层感知机
- 多层感知机:Multi-Layer Perceptron
- 多层感知机 Multi-Layer Perception Tensorflow实现
- Tensorflow实现多层感知机Multi-layer Preceptron
- 多层感知机及其BP算法(Multi-Layer Perception)
- tensorflow tutorials(六):用tensorflow实现多层感知器(Multilayer Perceptron)
- 多层感知机(Multilayer Perceptron)
- 多层感知机(Multi-Layer Perception)
- 线性分类模型--感知机(perceptron)
- 感知机(Perceptron)
- Perceptron(感知机)
- Tensorflow 实战 笔记 (一)实现多层感知机
- python实现感知机(perceptron)原型~
- 感知机(Perceptron)的python实现
- TensorFlow实现多层感知机
- tensorflow实现多层感知机
- TensorFlow实现多层感知机
- Wine trading in Gergovia UVA
- CentOS6.9下MySQL5.6编译安装
- Ajax的原理和应用
- django建站实录
- Android Studio自动检查代码规范并提示如何优化的一些插件
- 多层感知机(multi-layer perceptron)实现手写体分类(TensorFlow)
- 快速幂
- 基于以太坊实现代币发布
- 让MySql支持Emoji表情(MySQL中4字节utf8字符保存方法)
- Maven入门1-在Eclipse中新建Maven Web项目
- 博弈论
- Activiti工作流
- 黑盒测试方法二
- java8 collect操作函数列表