如何从 R 中的 sklearn 包中复制多标签二值化器的工作?

数据挖掘 r 多标签分类 混淆矩阵
2022-02-05 20:25:57

我想从 R 中的 sklearn.preprocessing 包中实现 MultiLabelBinarizer 的相同工作。我有每个示例(用于预测和实际)的标签列表,如下所示。

[[1]][1] "a"      "b"      "d"    "f"
[[2]][1] "g"      "b"      "f"    "d"
....

我想要的是把它放进去——

[[1 1 0 1 0 1 ...] [0 1 0 1 0 1 1 ..]]...

这样我就可以以 multilabel_confusion_matrix 的形式获得它。

1个回答

一种方法是将源代码从此处Python转换为R- 参见第 289 行

您还可以检查现有R包中的功能 - 例如mldrmlrcaret包。