Valueerror bei der Berechnung der Präzision mit Scikit-Learn

Question

from sklearn.metrics import precision_score 

precision_score(expected, predicted) 

erwartet, wo array([ 4., 3.]) ist

und vorhergesagt ist array([ 2., 4.])

ich folg bekommen. Fehler: *** ValueError: pos_label=1 is not a valid label: array([ 2., 3., 4.])

Wie kann dies behoben werden?

Answer 1

Sie benötigen den average Parameter für Multiklasse Etiketten.

sonst benötigen Sie pos_label als eine der Klasse Etiketten in beiden Arrays einzustellen d.h. 2, 3 oder 4:

>>> # score for all classes 
>>> precision_score(expected, predicted, average=None) 
array([ 0., 0., 0.]) 

>>> # score for each class 
>>> precision_score(expected, predicted, pos_label=2) 
0.0 
>>> precision_score(expected, predicted, pos_label=3) 
0.0 
>>> precision_score(expected, predicted, pos_label=4) 
0.0 

Referenz: sklearn.metrics.precision_score