Classification Scikit Learn Multilabel, Récupération d'étiquettes à partir de MultiLabelBinarizer

Question

Dans un problème de classification multilabel, j'utilise MultiLabelBinarizer pour transformer mes 20 étiquettes de texte en une liste binaire de zéros et de uns. Après la prédiction, j'obtiens ma liste de 20 valeurs binaires, et je voudrais sortir les étiquettes de texte correspondantes. Je me demande simplement si MultiLabelBinarizer() fournit une transformation de retour ou je devrais le faire manuellement.

Answer 1

Oui, MultiLabelBinarizer fournit une méthode nommée inverse_transform() qui convertira les étiquettes binarisées en noms d'origine qui lui ont été fournis pendant 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.