Entropía vs índice de Gini
En este ejercicio compararás la precisión del conjunto de pruebas de dt_entropy
con la precisión de otro árbol llamado dt_gini
. El árbol dt_gini
se entrenó en el mismo conjunto de datos utilizando los mismos parámetros, excepto el criterio de información, que se fijó en el índice de Gini utilizando la palabra clave 'gini'
.
X_test
, y_test
, dt_entropy
, así como accuracy_gini
, que corresponde a la precisión del conjunto de pruebas conseguida por dt_gini
, están disponibles en tu espacio de trabajo.
Este ejercicio forma parte del curso
Machine learning con modelos basados en árboles en Python
Instrucciones de ejercicio
- Importa
accuracy_score
desdesklearn.metrics
. - Predice las etiquetas del conjunto de prueba de
dt_entropy
y asigna el resultado ay_pred
. - Evalúa la precisión del conjunto de pruebas de
dt_entropy
y asigna el resultado aaccuracy_entropy
. - Revisa
accuracy_entropy
yaccuracy_gini
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Import accuracy_score from sklearn.metrics
from ____.____ import ____
# Use dt_entropy to predict test set labels
____= ____.____(____)
# Evaluate accuracy_entropy
accuracy_entropy = ____(____, ____)
# Print accuracy_entropy
print(f'Accuracy achieved by using entropy: {accuracy_entropy:.3f}')
# Print accuracy_gini
print(f'Accuracy achieved by using the gini index: {accuracy_gini:.3f}')