python - Python中标签二值化的逆过程

我已经为多类分类执行了标签二值化，它工作正常：

y_test
1
3
4
2
0

from sklearn.preprocessing import label_binarize

y_test_binarize = label_binarize(y_test, classes=[0, 1, 2, 3, 4])
y_test_binarize

0   1   0   0   0
0   0   0   1   0
0   0   0   0   1
0   0   1   0   0
1   0   0   0   0

接下来，我想做一个逆过程，从变量中得到y_test。
是否有任何预定义的方法？

最佳答案

您还可以使用LabelBinarizer，它将label_binarize函数包装在一个类中，并提供方法来转换为二进制数据，还可以将它们反向转换为原始类。

y_test = [1, 3, 4, 2, 0]

from sklearn import preprocessing
lb = preprocessing.LabelBinarizer()

y_test_binarize = lb.fit_transform(y_test)
#Output: y_test_binarize
array([[0   1   0   0   0],
       [0   0   0   1   0],
       [0   0   0   0   1],
       [0   0   1   0   0],
       [1   0   0   0   0]])

y_test_original = lb.inverse_transform(y_test_binarize)
#Output: y_test_original
array([1, 3, 4, 2, 0])

希望这有帮助。如果有什么问题可以问。