Scikit學習多標籤分類，從MultiLabelBinarizer獲取標籤

Question

在多標籤分類問題中，我使用MultiLabelBinarizer將我的20個文本標籤轉換爲零和一個二進制列表。

預測後，我得到20個二進制值的列表，我想輸出相應的文本標籤。

我只是想知道MultiLabelBinarizer（）是否提供了一個回頭轉換，或者我應該手動進行。

Answer 1

是的，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.