我可以手动创建混淆矩阵y_true数组吗？

问：

我不知道如何从 customdatagenerator 获取y_true

我可以直接创建数组并输入标签数据吗？

val_datagen = CustomDataGenerator(image_folders, label_folders, valid_dir, **params, shuffle = False)

Y_pred = model.predict(val_datagen)
y_pred = np.argmax(Y_pred, axis=1) 
y_true = np.array([0,0,0,0,0, 0,0,0,0,0, 0,0,   
                   1,1,1,1,1, 1,
                   2,2,2,2,2, 2,2,
                   3,3,3,3,3, 3,3,3,3,3, 3,3,3,3,3, 3,3,3,3,
                   4,4,4,4,
                   5,5,5,5,5, 5,5,5,5,5,
                   6,6,6,6,])
fig, ax = plt.subplots(figsize=(12,6))  
sns.heatmap(confusion_matrix(y_true, y_pred),annot=True, fmt="d", cmap='Greens',ax = ax)

在val_dataset我有 12 * label0、6 * label1、7 * label2.....

而“shuffle = False”确保y_true可以完美地对应y_pred