scikit-learn LabelBinarizer - 一个热门与虚拟编码

时间:2017-10-15 19:01:22

标签: python scikit-learn dummy-variable one-hot-encoding

scikit-learn LabelBinarizer中有一个很棒的功能,它的工作原理如下:

from sklearn.preprocessing import LabelBinarizer
lb = LabelBinarizer(sparse_output = True)
lb.fit_transform(['green', 'yellow', 'blue'])

输出:

[[0 1 0]
 [0 0 1]
 [1 0 0]]

但我想知道,如果我可以从它或任何获得以下输出 替代功能:

   [[0 0]
     [0 1]
     [1 0]]

我尝试了选项sparse_output = True,但这不是我想要的。

0 个答案:

没有答案