我试图用下面的代码为我的图像分类器绘制一个混淆矩阵,但我收到一条错误消息:“PrefetchDataset”对象没有属性“类”
Y_pred = model.predict(validation_dataset)
y_pred = np.argmax(Y_pred, axis=1)
print('Confusion Matrix')
print(confusion_matrix(validation_dataset.classes, y_pred)) # ERROR message generated
Run Code Online (Sandbox Code Playgroud)