python - 如何使用 pandas_ml 制作包含 2 个类的混淆矩阵？

Question

我正在尝试用绘图制作一个混淆矩阵，pandas_ml 似乎有一个函数，但它不适用于 2 个类。是否有一些 secret 选项可以使其正常工作？

from pandas_ml import ConfusionMatrix
ytrue = ['ham', 'ham', 'spam']
ypred = ['ham', 'spam', 'spam']
cm = ConfusionMatrix(ytrue, ypred)
cm

结果

Predicted  False  True  __all__
Actual                         
False          0     0        0
True           0     0        0
__all__        0     0        0

这个:

from pandas_ml import ConfusionMatrix
ytrue = ['ham', 'ham', 'spam', 'third']
ypred = ['ham', 'spam', 'spam', 'third']
cm = ConfusionMatrix(ytrue, ypred)
cm

结果

Predicted  ham  spam  third  __all__
Actual                              
ham          1     1      0        2
spam         0     1      0        1
third        0     0      1        1
__all__      1     2      1        4

Answer 1

也许为时已晚，但我也遇到了同样的问题。似乎当你有 pandas_ml 的 2 个类 ConfusionMatrix 时，你的输入必须是 bool 值。只需将“垃圾邮件”/“火腿”转换为 True/False 即可。

from pandas_ml import ConfusionMatrix
ytrue = np.array(['ham', 'ham', 'spam'])
ytrue = np.array(['ham', 'ham', 'spam'])
ypred = np.array(['ham', 'spam', 'spam'])
cm = ConfusionMatrix(np.where(ytrue == 'spam', True, False), np.where(ypred == 'spam', True, False))
cm