Sportmads Diario SPORT. Versión online del periódico deportivo. Todas las noticias
Cómo calcular el puntaje F1 en Scikit-learn para medir la precisión de tus modelos
Scikit-learn es una biblioteca ampliamente utilizada en aprendizaje automático y minería de datos. Esta biblioteca proporciona un conjunto de herramientas para la creación de modelos de aprendizaje automático y su evaluación. Uno de los métodos comúnmente utilizados en la evaluación de modelos de aprendizaje automático es el puntaje F1.
El puntaje F1 es una medida de la precisión de un modelo de aprendizaje automático. Es una combinación de las medidas de precisión y recuperación. La precisión es la fracción de la recuperación relevante de todas las recuperaciones, mientras que la recuperación es la fracción de los elementos relevantes recuperados de todos los elementos relevantes.
La fórmula para el puntaje F1 es la siguiente:
F1 = 2 * (precisión * recuperación) / (precisión + recuperación)
El puntaje F1 varía de cero a uno, donde un puntaje de uno indica una precisión perfecta y un puntaje de cero indica una precisión nula.
Para calcular el puntaje F1 en Scikit-learn, necesitamos importar la función “f1_score” desde la biblioteca Scikit-learn. La función “f1_score” toma dos parámetros: las etiquetas verdaderas y las etiquetas predichas.
Por ejemplo, supongamos que tenemos un conjunto de datos de entrenamiento y queremos calcular el puntaje F1 de un modelo de aprendizaje automático. Primero, ajustamos nuestro modelo a los datos de entrenamiento y predecimos las respuestas para los datos de prueba. Luego, podemos calcular el puntaje F1 utilizando la siguiente línea de código:
from sklearn.metrics import f1_score
f1_score(y_true, y_pred)
Donde y_true son las etiquetas verdaderas y y_pred son las etiquetas predichas.
Además, es importante mencionar que podemos especificar el parámetro “average” en la función “f1_score” para calcular el puntaje F1 para diferentes casos. Por ejemplo, el valor predeterminado del parámetro “average” es “binario”, lo que significa que el puntaje F1 se calcula para cada clase de manera individual.
En resumen, el puntaje F1 es una medida importante en la evaluación de modelos de aprendizaje automático. Scikit-learn proporciona una función fácil de usar para calcular el puntaje F1 para diferentes casos. Al tener en cuenta tanto la precisión como la recuperación, el puntaje F1 es una medida apropiada para evaluar la calidad de un modelo de aprendizaje automático en muchos escenarios.
f1 score scikit learn
