gpt4 book ai didi

python - 从 keras 生成器获取真实标签

转载 作者:行者123 更新时间:2023-12-03 23:53:18 25 4
gpt4 key购买 nike

我想用 sklearn.metrics.confusion_matrix(y_true, y_pred) 为 keras 模型创建混淆矩阵。

训练模型后,我可以使用 predict_generator(generator) 获得对测试数据集的预测,这给了我 y_pred .怎样才能得到对应的真标签,y_true从数据生成器?

最佳答案

generator.classes将以稀疏格式为您提供观察值。您可能需要密集的(即单热编码格式)。你可以用:

import pandas as pd
pd.get_dummies(pd.Series(generator.classes)).to_dense()

但请注意:您必须设置生成器的 shuffle归因于 False在生成预测并获取观察到的类之前,否则您的预测和观察将不会对齐!

关于python - 从 keras 生成器获取真实标签,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53949991/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com