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
,但这不是我想要的。