小编zZz*_*ZzZ的帖子

如何在 Tensorflow 中为预取数据集绘制混淆矩阵

我试图用下面的代码为我的图像分类器绘制一个混淆矩阵,但我收到一条错误消息:“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)

python machine-learning image-processing keras tensorflow

6
推荐指数
2
解决办法
5389
查看次数