首页 > 解决方案 > Keras:带有 predict_generator 的混淆矩阵

问题描述

我没有使用“ImageDataGenerator”,因为我使用的是 hdf5 文件。我使用 DataGenerator 类 (1) 将数据提供给“model.fit_generator”。那么,鉴于我有一批随机数据,我怎样才能得到混淆矩阵呢?请注意,我知道如何使用 sklearn 使用“model.predict”来获取它。

(1) https://stanford.edu/~shervine/blog/keras-how-to-generate-data-on-the-fly#

标签: tensorflowkeras

解决方案


推荐阅读