TensorFlow 2.0全连接单隐藏层网络建模实现

1、载入数据

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

print("Tensorflow版本是:",tf.__version__)

mnist = tf.keras.datasets.mnist
(train_images,train_labels),(test_images,test_labels)=mnist.load_data()

2、数据集划分

total_num = len(train_images)
valid_split = 0.2   #验证集的比例占20%
train_num = int(total_num*(1-valid_split))#训练集的数目

train_x = train_images[:train_num]#前部分给训练集
train_y = train_labels[:train_num]

valid_x = train_images[train_num:]#后20%给验证集
valid_y = train_labels[train_num:]

test_x = test_images
test_y = test_labels

3、数据塑性

train_x = train_x.reshape(-1,784)
val