从predict_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]

您正在用生成器编写float数组的元素。

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