Untitled
unknown
python
3 years ago
1.4 kB
10
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...