Skip to content

fixing binarizer for 2 class labels

Maggie Chen requested to merge meiqi/umami:maggiechen_label_binarize_fix into master

Fixing the issue with label_binarize from sklearn. When there are only 2 classes for classification, label_binarize returns one single column vector instead of one column per class. The fix adds a dummy class label i.e -1, and removes the last column after the binarizer.

Edited by Maggie Chen

Merge request reports