人工智能 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)

原文地址:https://www.cnblogs.com/imlvbu/p/7754650.html