利用Tensorflow2.0、softmax实现手写字体识别

参考链接
项目采用Jupyter notebook编写

import tensorflow as tf
from tensorflow.keras.datasets import mnist
import numpy as np
from sklearn.preprocessing import StandardScaler

print(tf.__version__)

2.0.0

##加载数据     60000条训练集   10000条测试集  
(x_train_all, y_train_all), (x_test, y_test) = mnist.load_data()  #此处会去官网加载数据，可能比较慢
print(type(x_train_all))

<class 'numpy.ndarray'>

#print((x_train.shape),(x_test.shape))   #(60000, 28, 28) (10000, 28, 28)
#数据归一化
scaler = StandardScaler()
scaled_x_train_all = scaler.fit_transform(x_train_all.astype(np.float32).reshape(-1,1)).reshape(-1,28,28)
scaled_x_test = scaler.transform(x_test.astype(np.float32).reshape