F1_Score: F1 Score

View source: R/Classification.R

F1_ScoreR Documentation

F1 Score

Description

Compute the F1 Score.

Usage

F1_Score(y_true, y_pred, positive = NULL)

Arguments

y_true

Ground truth (correct) 0-1 labels vector

y_pred

Predicted labels vector, as returned by a classifier

positive

An optional character string for the factor level that corresponds to a "positive" result

Value

F1 Score

Examples

data(cars)
logreg <- glm(formula = vs ~ hp + wt,
              family = binomial(link = "logit"), data = mtcars)
pred <- ifelse(logreg$fitted.values < 0.5, 0, 1)
F1_Score(y_pred = pred, y_true = mtcars$vs, positive = "0")
F1_Score(y_pred = pred, y_true = mtcars$vs, positive = "1")

MLmetrics documentation built on May 29, 2024, 9:34 a.m.