如何得到Tensorflow-hub模型的混淆矩阵

我已经通过培训培训了一个基于迁移学习的模型。我已经在很多地方寻找关于生成混淆矩阵的提示，但我还没有找到正确的解决方案

有人知道这是否可能吗

我最后一次尝试是将结果写在Excel表格中，但在Excel中找不到混淆矩阵的多类计算公式

---

任何帮助都会很好

您可以尝试使用tf.math.conflusion\u matrix函数。
它根据预测和标签计算混淆矩阵。
看

示例：

my_confusion_matrix=tf.math.confusion_matrix（标签=[1,2,4]，预测=[2,2,3]）
使用tf.Session（）作为sess：
打印（sess.run（我的矩阵））
#印刷品#
[[0 0 0 0 0]
[0 0 1 0 0]
[0 0 1 0 0]
[0 0 0 0 0]
[0 0 0 1 0]]
#假设标签为[0,1,2,3,4]，从而导致5X5混淆矩阵

关于混淆矩阵的计算，TF（或TF Hub）没有任何特定内容。一旦得到数据的预测，只需进行任何“标准”混淆矩阵计算（例如，对于python），将标签和预测输入其中，然后将矩阵输出。