使用multilabel_binarizer后如何从列号获取类名?

时间:2019-11-19 18:48:40

标签: multilabel-classification

这是我编写的代码。执行后,y包含100个列,这些列表示类,其中数据点在该类中,值为1。

multilabel_binarizer = MultiLabelBinarizer()
multilabel_binarizer.fit(meta['Tags'])

# transform target variable
y = multilabel_binarizer.transform(meta['Tags'])

我想获得一个表示每一列代表的类名的数组。我知道逆变换用于此目的,但我无法弄清楚语法。如果还有其他方法,请分享

1 个答案:

答案 0 :(得分:2)

您可以使用以下功能-

Classes= multilabel_binarizer.classes_