Scikit了解Multilabel分类,从MultiLabelBinarizer获取标签

时间:2017-04-06 11:47:14

标签: scikit-learn multilabel-classification

在多标签分类问题中,我使用MultiLabelBinarizer将我的20个文本标签转换为零和1的二进制列表。

预测后,我得到了20个二进制值的列表,我想输出相应的文本标签。

我只是想知道MultiLabelBinarizer()是否提供了回归转换,或者我应该手动进行转换。

1 个答案:

答案 0 :(得分:4)

是的,MultiLabelBinarizer提供了一个名为inverse_transform()的方法,它会将二值化标签转换回fit()期间提供给它的原始名称。

http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.MultiLabelBinarizer.html#sklearn.preprocessing.MultiLabelBinarizer.inverse_transform

  

inverse_transform(YT)

Transform the given indicator matrix into label sets
Parameters:   

yt : array or sparse matrix of shape (n_samples, n_classes)

    A matrix containing only 1s ands 0s.

Returns:  

y : list of tuples

    The set of labels for each sample such that y[i] consists of classes_[j] for each yt[i, j] == 1.