Untitled

 avatar
unknown
python
3 years ago
1.4 kB
6
Indexable
## кортика
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix
import pandas as pd

## функция матрицы ошибок
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay


## а как присвоить фаийл?
# Что сравнивать
actual_data = \
       ['apples',  'pears',   'apples',
        'apples',  'apples',  'pears',
        'oranges', 'oranges', 'apples',
        'apples',  'apples',  'apples',
        'apples',  'apples',  'pears',
        'apples',  'oranges', 'apples',
        'apples',  'apples']

## С чем сравнивать

predicted_data = \
      ['oranges', 'pears',   'apples',
       'apples',  'apples',  'pears',
       'oranges', 'oranges', 'apples', 
       'apples',  'apples',  'apples',
       'apples',  'apples',  'pears',
       'apples',  'oranges', 'oranges',
       'apples',  'oranges']
    
## Делаем матрицу
cum = confusion_matrix(actual_data, predicted_data)

## Выводим
print(cum)

## Тут картинки пошли поехали
cmd_obj = ConfusionMatrixDisplay(cm, display_labels=['apples', 'oranges', 'pears'])

cmd_obj.plot()

cmd_obj.ax_.set(
                title='Sklearn Confusion Matrix with labels!!', 
                xlabel='PREDICTED', 
                ylabel='Actual')

plt.show()
Editor is loading...