首页 > 技术文章 > 人工智能 tensorflow框架-->Softmax回归模型的python实现 08

imlvbu 2017-10-30 15:22 原文

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)

推荐阅读