如何在Keras中打印多类图像分类的y_pred值

Question

如何在Keras中打印多类图像分类的y_pred值？我还想打印用于多类分类的混淆矩阵。我的代码如下：

train, val = train_test_split(index, test_size=0.2,random_state=42,stratify=index["level"])
train_datagen = ImageDataGenerator(rescale            = 1/255,
shear_range        = 0.2,
zoom_range         = 0.2,
horizontal_flip    = True,
rotation_range     = 40,
width_shift_range  = 0.2,
height_shift_range = 0.2)
val_datagen = ImageDataGenerator(rescale = 1/255)
train_set = train_datagen.flow_from_dataframe(
train,
path,
x_col="image_name",
y_col="level",
class_mode="raw",
color_mode="rgb",
batch_size=32,
target_size=(64, 64))
val_set = val_datagen.flow_from_dataframe(
val,
path,
x_col="image_name",
y_col="level",
class_mode="raw",
color_mode="rgb",
batch_size=32,
target_size=(64, 64))
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])

Answer 1

定义模型并在列车数据上对其进行训练后，要返回基于高概率模型预测的预测类，请使用此方法np.argmax(model_predictions，axis=-1(，甚至可以使用train_set.class_indices 定义其标签