Visualizar la diferencia entre 2 clasificadores

Question

He entrenado 2 clasificadores binarios con los mismos datos (un Árbol de Decisión y un Bosque Aleatorio).

Ambos hicieron una predicción sobre los mismos datos de prueba.

Ahora, quiero visualizar la diferencia de clasificación entre estos 2 clasificadores. (usando Python)

Ya se me ha ocurrido esto:

x = #[List of IDs]
y1 = #[List of predictions from Decision Tree]
y2 = #[List of predictions fro Random Forest]

plt.scatter(x, y1, color='r')
plt.scatter(x, y2,  color='b')
plt.xlim(xmin=890, xmax=1300)
plt.ylim(ymin=-1, ymax=2)
plt.yticks(np.arange(2), ('Class 0', 'Class 1'))
plt.xlabel('ID')
plt.ylabel('Class')
plt.show()

El resultado es el siguiente:

Como puede verse, la trama no está muy clara.

¿Alguna sugerencia sobre cómo puedo mejorar esta trama? ¿O una forma diferente de trazar este concepto?

Answer 1

Una buena forma de comparar clasificadores es representar gráficamente el Curva ROC . Es bastante fácil de codificar y ofrece una visión completa del rendimiento de los clasificadores para todos los umbrales diferentes.