Desenho de um gráfico de mosaico da matriz de confusão
Embora o cálculo da matriz de desempenho possa ser divertido, ele se tornaria tedioso se você precisasse de várias matrizes de confusão de modelos diferentes. Felizmente, o método .pred_table() pode calcular a matriz de confusão para você.
Além disso, você pode usar a saída do método .pred_table() para visualizar a matriz de confusão, usando a função mosaic().
churn e mdl_churn_vs_relationship estão disponíveis.
Este exercício faz parte do curso
Introdução à regressão com statsmodels em Python
Instruções do exercício
- Importe a função
mosaic()destatsmodels.graphics.mosaicplot. - Crie o site
conf_matrixusando o método.pred_table()e imprima-o. - Desenhe um gráfico em mosaico da matriz de confusão.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Import mosaic from statsmodels.graphics.mosaicplot
____
# Calculate the confusion matrix conf_matrix
conf_matrix = ____
# Print it
print(conf_matrix)
# Draw a mosaic plot of conf_matrix
____
plt.show()