(P5)使用keras进行多分类问题(2)-独热编码（onehot）

全部代码

import tensorflow as tf
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

(train_image, train_label), (test_image, test_label) = tf.keras.datasets.fashion_mnist.load_data()

#进行归一化处理
train_image = train_image/255
test_image = test_image/255

# 将label转化为onehot的编码形式
train_label_onehot = tf.keras.utils.to_categorical(train_label)
# plt.imshow(train_image[3])
# plt.show()
# print(train_label[3], train_label_onehot[3])
test_label_onehot = tf.keras.utils.to_categorical(test_label)
# print(test_label_onehot)

model = tf.keras.Sequential()
model.add(tf.keras.layers.Flatten(input_shape=(28,28)))
model.add(tf.keras.layers