首页 > 代码库 > TensorFlow经典案例4:实现logistic回归
TensorFlow经典案例4:实现logistic回归
#TensorFlow实现Logistic 回归 import tensorflow as tf #导入手写数字集 from tensorflow.examples.tutorials.mnist import input_data mnist = input_data.read_data_sets("/tmp/data/", one_hot=True) #学习参数 learning_rate = 0.01 training_epoches = 25 batch_size = 100 display_step = 1 #构造图 x = tf.placeholder(tf.float32,[None,784]) y = tf.placeholder(tf.float32,[None,10]) W = tf.Variable(tf.zeros([784,10])) b = tf.Variable(tf.zeros([10])) prediction = tf.nn.softmax(tf.matmul(x,W) + b) cost = tf.reduce_mean(-tf.reduce_sum(y*tf.log(prediction),reduction_indices=1)) optimizer = tf.train.GradientDescentOptimizer(learning_rate).minimize(cost) init = tf.global_variables_initializer() with tf.Session() as sess: sess.run(init) for epoch in range(training_epoches): avg_cost = 0. total_batch = int(mnist.train.num_examples/batch_size) for i in range(total_batch): batch_xs,batch_ys = mnist.train.next_batch(batch_size) _,c =sess.run([optimizer,cost],feed_dict={x:batch_xs,y:batch_ys}) avg_cost += c / total_batch if (epoch+1) % display_step == 0: print("Epoch:","%0.4d" %(epoch+1),"cost","{:.9f}".format(avg_cost)) print("训练结束") correct_prediction = tf.equal(tf.argmax(prediction,1),tf.argmax(y,1)) accuracy = tf.reduce_mean(tf.cast(correct_prediction,tf.int32)) print("Accuracy:", accuracy.eval({x: mnist.test.images, y: mnist.test.labels}))
TensorFlow经典案例4:实现logistic回归
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。