您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

混淆矩阵不支持Multilabel-indicator

混淆矩阵不支持Multilabel-indicator

不,您输入的内容confusion_matrix必须是预测列表,而不是OHE(一种热编码)。调用argmax您的y_testy_pred,您应该得到期望的结果。

confusion_matrix(
    y_test.values.argmax(axis=1), predictions.argmax(axis=1))

array([[1, 0],
       [0, 2]])
其他 2022/1/1 18:25:08 有420人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶