Tensorflow基础代码报错学习笔记11——classification分类学习

原教程地址

原代码

更换了tensorflow1.0版本之后代码跟着up主的教程敲就可以了，这里面没什么需要改动的

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
# 如果电脑中没有数据集，会自动下载
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)

def add_layer(inputs, in_size, out_size, activation_function=None):
    # add one more layer and return the output of this layer
    Weights = tf.Variable(tf.random.normal([in_size, out_size]))
    biases = tf.Variable(tf.zeros([1, out_size]) + 0.1)
    Wx_plus_b = tf.matmul(inputs, Weights) + biases
    if activation_function is