将归一化混淆矩阵限制为2个小数点

时间:2019-11-26 11:18:05

标签: python-3.x

我正在显示标准化的混淆矩阵,如下所示:

[[0.9        0.         0.0106383  0.         0.        ]
[0.         0.83673469 0.05319149 0.01801802 0.04347826]
[0.         0.04081633 0.87234043 0.09009009 0.        ]
[0.         0.         0.14893617 0.83783784 0.17391304]
[0.         0.         0.04255319 0.04504505 0.60869565]]

我只想显示两位小数。

这是我的标准化混淆矩阵代码:

# Confusion Matrix
conf_mat = confusion_matrix(y_test, predicted)
conf_mat_norm = conf_mat / conf_mat.astype(np.float).sum(axis=1))

1 个答案:

答案 0 :(得分:0)

您可以使用此方法,它对我有用。如果适合您,请对该答案进行投票。

conf_mat_norm= np.around(conf_mat.astype('float') / conf_mat.sum(axis=1)[:, np.newaxis], decimals=2)