我正在显示标准化的混淆矩阵,如下所示:
[[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))
答案 0 :(得分:0)
您可以使用此方法,它对我有用。如果适合您,请对该答案进行投票。
conf_mat_norm= np.around(conf_mat.astype('float') / conf_mat.sum(axis=1)[:, np.newaxis], decimals=2)