ComenzarEmpieza gratis

Predicciones de clasificación

En la validación de modelos, a menudo es importante saber más sobre las predicciones que solo la clasificación final. Al predecir quién ganará un partido, la mayoría también quiere saber con qué probabilidad ganará un equipo.

Probability Prediction Meaning
0 < .50 0 Team Loses
.50 + 1 Team Wins

En este ejercicio, vas a usar los métodos .predict() y .predict_proba() con el conjunto de datos tic_tac_toe. El primer método te dará una predicción de si el Jugador Uno ganará la partida, y el segundo método te proporcionará la probabilidad de que el Jugador Uno gane. Usa rfc como el modelo de clasificación de random forest.

Este ejercicio forma parte del curso

Validación de modelos en Python

Ver curso

Instrucciones del ejercicio

  • Crea dos arrays de predicciones: uno para los valores de la clasificación y otro para las probabilidades predichas.
  • Usa el método .value_counts() de una Series de pandas para imprimir el número de observaciones asignadas a cada clase.
  • Imprime la primera observación de probability_predictions para ver cómo están estructuradas las probabilidades.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# Fit the rfc model. 
rfc.fit(X_train, y_train)

# Create arrays of predictions
classification_predictions = rfc.____(X_test)
probability_predictions = rfc.____(X_test)

# Print out count of binary predictions
print(pd.Series(____).____())

# Print the first value from probability_predictions
print('The first predicted probabilities are: {}'.format(____[____]))
Editar y ejecutar código