tensorflow深度学习-mnist数据集读入-初试

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
import tensorflow as tf
from tensorflow import keras
from  tensorflow.keras import layers
import numpy as np
import matplotlib.pyplot as plt

# from tensorflow.keras.datasets import mnist
# mnist is the handwriting number dataset 0-9

def load_mnist(path):
    f=np.load(path)
    x_train, y_train = f['x_train'], f['y_train']
    x_test, y_test = f['x_test'], f['y_test']
    f.close()
    return (x_train, y_train), (x_test, y_test)


(x_train, y_train), (x_test, y_test) = load_mnist("......")
print(x_train.shape)
print(y_train.shape)


N=3
idx=0
for pages in range(5):
    for i in range(N):
        for j in range(N):
            num=i*N+j
            plt.subplot(N,N,num+1)
            plt.imshow(x_train[num+idx],cmap=plt.get_cmap('gray'))
    idx+=N*N
    # show the plot
    plt.show()

网络还没有搭建，先就读入看看效果

输出了五张3*3手写数字的图片，还阔以