如何使用keras模型和TFrecords数据制作混淆矩阵？

Question

我训练了一个使用TFrecords的keras模型，因为训练数据和验证数据不适合内存。现在我想做一个混淆矩阵，但是使用numpy数组或“内存中”张量的经典方法不起作用。（以tf.math.conflusion\u矩阵为例）

有什么方法可以简单地从TFrecords中创建混淆矩阵吗？即使数据量巨大，制作混乱矩阵也不应该占用太多内存。（回答“不，因为……”对我也很有帮助。）

或者我只需要处理较小的验证数据