Вес многометочного вычислительного класса — нехэшируемый тип ⇐ Python

[Anonymous]

Работа над задачой классификации с несколькими метками с 13 возможными выходными данными в моей нейронной сети с помощью Keras, sklearn и т. д.< /p>

Каждый вывод может представлять собой массив типа [0, 0, 1, 0, 1, 1, 0, 0, 0, 0, 0, 1,0].< /p>

У меня есть набор данных о дисбалансе, и я пытаюсь применить метод compute_class_weight, например:

Код: Выделить всё

class_weight = compute_class_weight('balanced', np.unique(Y_train), Y_train)

Когда я пытаюсь запустить свой код, я получаю Unhashable Type: 'numpy.ndarray':

Код: Выделить всё

Traceback (most recent call last):
File "main.py", line 115, in 
train(dataset, labels)
File "main.py", line 66, in train
class_weight = compute_class_weight('balanced', np.unique(Y_train), Y_train)
File "/home/python-env/env/lib/python3.6/site-packages/sklearn/utils/class_weight.py", line 41, in compute_class_weight
if set(y) - set(classes):
TypeError: unhashable type: 'numpy.ndarray'

Я знаю, что это потому, что я работаю с массивами и уже пытался добавить немного слов,

т.е.:

Код: Выделить всё

class_weight_dict = dict(enumerate(np.unique(y_train), class_weight))

Ну, я не знаю, что делать, пробовал другие стратегии, но безуспешно...
Есть идеи?

Заранее спасибо!

Подробнее здесь: https://stackoverflow.com/questions/548 ... hable-type