MultiLabelBinarizer:inverse_transform在单个样本上失败?

时间:2019-04-06 20:01:47

标签: scikit-learn multilabel-classification

我想将inverse_transform的MultiLabelBinarizer应用于单个样本,例如:

labels = ['Architektur & Garten',
          'Ganzheitliches Bewusstsein',
          'Glaube & Ethik',
          'Kinderbuch & Jugendbuch',
          'Künste',
          'Literatur & Unterhaltung',
          'Ratgeber',
          'Sachbuch']

samples = []
for l in labels:
   samples.append([l])

from sklearn.preprocessing import MultiLabelBinarizer
m = MultiLabelBinarizer()
m.fit_transform(samples)

如果我现在将MultiLabelBinarizer应用于矩阵,它将起作用:

s = np.array([[0, 1, 0, 0, 0, 0, 0, 0],[0, 0, 0, 0, 0, 0, 0, 1]])
m.inverse_transform(s)
[('Ganzheitliches Bewusstsein',), ('Sachbuch',)]

但是,如果我尝试将其应用于单个样本(即向量),则会失败:

import numpy as np
s = np.array([0, 1, 0, 0, 0, 0, 0, 0])
m.inverse_transform(s)

--> 957         if yt.shape[1] != len(self.classes_):
    958             raise ValueError('Expected indicator for {0} classes, but got {1}'
    959                              .format(len(self.classes_), yt.shape[1]))

1 个答案:

答案 0 :(得分:1)

从您的评论来看,好像您已经解决了它。如果这对其他人有帮助,请提供一些详细信息:

在第一个示例中,如果我们打印s的尺寸,则会得到(2, 8)

>>> s = np.array([[0, 1, 0, 0, 0, 0, 0, 0],[0, 0, 0, 0, 0, 0, 0, 1]])
>>> s.shape
(2,8)

在第二个示例中,如果我们做同样的事情,我们得到(8,)

>>> s = np.array([0, 1, 0, 0, 0, 0, 0, 0])
>>> s.shape
(8,)

问题出在第二个示例上,抛出错误inverse_transform有助于表明它希望yt.shape[1]可用。在第二个示例中,该尺寸不可用,因此会出现错误。

这可以通过两种方式解决:

  1. 只需添加方括号:s = np.array([0, 1, 0, 0, 0, 0, 0, 0])变成s = np.array( [ [0, 1, 0, 0, 0, 0, 0, 0] ] )
  2. 重塑:s = np.reshape(s, (1, s.shape[0]))

无论哪种方式,之后s.shape的输出将是(1, 8),而m.inverse_transform(s)将按预期工作。