2012-12-12 1 views
2

Je voudrais savoir s'il est possible d'obtenir plus de nombres après la virgule avec classification_report de sklearn (scikit).Précision de sklearn.metric classification_report

atm il ressemble à ceci:

  precision recall f1-score support 

     1  0.61  0.73  0.67  71194 
     2  0.64  0.33  0.43  13877 
     3  0.56  0.59  0.57  61591 
     4  0.64  0.51  0.57  13187 
     5  0.66  0.69  0.67  57530 
     6  0.54  0.06  0.11  2391 
     7  0.54  0.40  0.46  30223 

avg/total 0,60 0,60 0,60 249993

Je ne pense pas qu'il est possible avec cette méthode, mais peut-être quelqu'un avait le même idée (probablement).

Je sais que sklearn.metrics.precision_score existe, bien que le classement_report soit une manière agréable d'afficher tous les résultats à la fois.

Répondre

5

Non possible selon le source code. Voir les lignes 819 et 830, les chaînes de format sont codées en dur à %0.2f. Si vous le voulez vraiment, il suffit de le changer dans votre fichier local sklearn/metrics/metrics.py. Mieux encore, ajoutez un argument à classification_report avec un nombre de précision et utilisez-le. Et soumettez votre patch au projet!