从 predict_proba() 中检索非零 class 概率

Question

我只想检索非零 class 概率。我下面的代码不断产生以下错误

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]

您正在使用生成器重写浮点数组的元素。

如果删除[:]，您可以将其存储为生成器，但不允许建立索引。因此尝试使用列表（方括号）。