Список Label_binarize без [] - PullRequest
Новая
       29

Список Label_binarize без []

0 голосов
/ 21 марта 2020

Я хочу преобразовать метки классов из y_train (type = list) в двоичные значения. Это работает при использовании для l oop с оператором if: 

y_train_bin = []
for i in y_list:
   if i == 'label1'
      i = 0
      y_train_bin.append(i)
   else:
      i = 1
      y_train_bin.append(i)

Когда я использую: 

y_train_bin = sklearn.preprocessing.label_binarize(y_list, ['label1','label2']

, я получаю массив numpy .array с [[0] , [1], [1], ....., [0]] как результат.

Этот тип, однако, нельзя использовать с sklearn.metrics.roc_curve. Есть ли способ все еще использовать label_binarize для этого?

...