官术网_书友最值得收藏!

Visualizing the training data

The following function will help you visualize the MNIST data. By passing in the index of a training example, the show_digit function will display that training image along with its corresponding label in the title:

# Visualize the data
import matplotlib.pyplot as plt
%matplotlib inline

#Displaying a training image by its index in the MNIST set
def display_digit(index):
    label = y_train[index].argmax(axis=0)
    image = X_train[index]
    plt.title('Training data, index: %d,  Label: %d' % (index, label))
    plt.imshow(image, cmap='gray_r')
    plt.show()
    
# Displaying the first (index 0) training image
display_digit(0)
X_train = X_train.reshape(60000, 784)
X_test = X_test.reshape(10000, 784)
X_train = X_train.astype('float32')
X_test = X_test.astype('float32')
X_train /= 255
X_test /= 255
print("Train the matrix shape", X_train.shape)
print("Test the matrix shape", X_test.shape)

#One Hot encoding of labels.
from keras.utils.np_utils import to_categorical
print(y_train.shape)
y_train = to_categorical(y_train, 10)
y_test = to_categorical(y_test, 10)
print(y_train.shape)
主站蜘蛛池模板: 洞头县| 刚察县| 吴桥县| 邯郸市| 崇信县| 咸宁市| 铜陵市| 朝阳县| 崇信县| 博爱县| 合水县| 巴南区| 正镶白旗| 峨眉山市| 南漳县| 寻甸| 大城县| 昆山市| 衡山县| 从江县| 杭锦旗| 晋江市| 岑巩县| 柯坪县| 宝坻区| 灌云县| 高邮市| 平湖市| 张家港市| 大理市| 白朗县| 普安县| 鄢陵县| 卓资县| 凤冈县| 宜昌市| 紫金县| 颍上县| 甘德县| 汨罗市| 阳泉市|