我已经为多类分类执行了标签二值化,并且工作正常:
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
Run Code Online (Sandbox Code Playgroud)
接下来,我想做一个逆过程来y_test从y_test_binarize变量中获取。
有没有预定义的方法?