Récupérer les probabilités de classe non nulles de predict_proba()

Question

Je souhaite récupérer uniquement les probabilités de classe non nulles. Mon code ci-dessous génère l'erreur suivante

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

Essayez ceci !

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

vous avez écrit les éléments d'un tableau de flottants avec un générateur.

Si vous supprimez [:] vous pouvez le stocker en tant que générateur, mais il ne permettra pas l'indexation. Essayez donc d'utiliser une liste (accolades carrées).