人工智能 tensorflow框架-->Softmax回归模型的python实现 08

1.导入所需模块：

import tensorflow as tf
import numpy as np

 

2.导入mnist数据集：

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)

 

3.定义一个占位符，用来表示输入任意数量的mnist图像

x = tf.placeholder("float", [None, 784])

 

4.定义权重值 和 偏置量

w = tf.Variable(tf.zeros([784,10]))     
b = tf.Variable(tf.zeros([10]))            

 

5.softmax模型代码

y = tf.nn.softmax(tf.matmul(x,w) + b)