Python 预训练张量流模型的混淆矩阵

我训练了tensorflow的mobilenet v1模型来检测我自己的类，但是我需要从生成的文件中获取模型结果的混淆矩阵，但是我找不到任何对我创建tensorflow模型矩阵有用的帖子。

试试这个

import matplotlib.pyplot as plt
from matplotlib.pyplot import imshow
from sklearn.metrics import confusion_matrix, classification_report
y_pred=[]
preds= model.predict etc.....
for i, p in enumerate (preds):
    p=pred[i]
    predicted_class_index=np.argmax(p)  # this the the predicted column with highest probability assuming you class_mode='categorical'
    y_pred.append(predicted_class_index) # this is a list of predictions
# now you need to create a list of the corresponding true labels
# how you get this depends on how you supplied the data to model.fit
# but somewhere you should have a list of labels - these are y_true
# then do
cm = confusion_matrix(y_true, y_pred )

如果使用来自目录的train\u gen=ImageDataGenerator.flow\u并设置shuffle=False，则y\u true=train\u gen.labels