Criando uma matriz de confusão no python
Para criar uma matriz de confusão no python, usaremos as bibliotecas do scikit-learn, para tanto, precisaremos instalar.
pip install scikit-learn
Segue abaixo o código, onde faremos a matriz de confusão.
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import seaborn as sns
# Valores reais (verdadeiros)
y_true = [2, 0, 2, 2, 0, 1]
# Valores previstos pelo modelo
y_pred = [0, 0, 2, 2, 0, 2]
# Gerar a matriz de confusão
conf_matrix = confusion_matrix(y_true, y_pred)
# Imprimir a matriz de confusão
print("Matriz de Confusão:")
print(conf_matrix)
# Para uma visualização melhor, você pode usar a biblioteca Seaborn para plotar a matriz de confusão
plt.figure(figsize=(10,7))
sns.heatmap(conf_matrix, annot=True, fmt="d")
plt.xlabel('Valores Previstos')
plt.ylabel('Valores Verdadeiros')
plt.title('Matriz de Confusão')
plt.show()
No código acima, podemos ver dois vetores, y_true, y_pred, onde os valores obtidos na leitura dos itens estão no vetor y_pred.
A matriz de referencia, é posta no y_true.
Por fim criamos a matriz de confusão:
conf_matrix = confusion_matrix(y_true, y_pred)
Criação da matriz de confusão