Извлечь ненулевые вероятности класса изgnastt_proba ()

Question

Я хочу получить только ненулевые вероятности класса. Мой код ниже продолжает генерировать следующую ошибку

print(clf.predict(xtest))
pp = clf.predict_proba(xtest)[0] 
pp[:] = ([ind,value] for ind,value in enumerate(pp) if value > 0)

for ind,val in enumerate(pp):
    print('\t',clf.classes_[pp[ind][0]],'->',pp[ind][1])
print('\n\n\n\n')

Answer 1

Попробуйте!

pp = clf.predict_proba(xtest)[0] 
pp = [[ind,value] for ind,value in enumerate(pp) if value > 0]

вы закончили писать элементы массива с плавающей запятой генератором.

Если вы удалите [:], вы можете сохранить его как генератор, но он не разрешит индексирование. Поэтому попробуйте использовать список (квадратные скобки).