These functions calculate the sens()
(sensitivity) of a measurement system
compared to a reference result (the "truth" or gold standard).
Highly related functions are spec()
, ppv()
, and npv()
.
sens(data, ...)# S3 method for data.frame
sens(
data,
truth,
estimate,
estimator = NULL,
na_rm = TRUE,
case_weights = NULL,
event_level = yardstick_event_level(),
...
)
sens_vec(
truth,
estimate,
estimator = NULL,
na_rm = TRUE,
case_weights = NULL,
event_level = yardstick_event_level(),
...
)
sensitivity(data, ...)
# S3 method for data.frame
sensitivity(
data,
truth,
estimate,
estimator = NULL,
na_rm = TRUE,
case_weights = NULL,
event_level = yardstick_event_level(),
...
)
sensitivity_vec(
truth,
estimate,
estimator = NULL,
na_rm = TRUE,
case_weights = NULL,
event_level = yardstick_event_level(),
...
)
A tibble
with columns .metric
, .estimator
,
and .estimate
and 1 row of values.
For grouped data frames, the number of rows returned will be the same as the number of groups.
For sens_vec()
, a single numeric
value (or NA
).
Either a data.frame
containing the columns specified by the
truth
and estimate
arguments, or a table
/matrix
where the true
class results should be in the columns of the table.
Not currently used.
The column identifier for the true class results
(that is a factor
). This should be an unquoted column name although
this argument is passed by expression and supports
quasiquotation (you can unquote column
names). For _vec()
functions, a factor
vector.
The column identifier for the predicted class
results (that is also factor
). As with truth
this can be
specified different ways but the primary method is to use an
unquoted variable name. For _vec()
functions, a factor
vector.
One of: "binary"
, "macro"
, "macro_weighted"
,
or "micro"
to specify the type of averaging to be done. "binary"
is
only relevant for the two class case. The other three are general methods
for calculating multiclass metrics. The default will automatically choose
"binary"
or "macro"
based on estimate
.
A logical
value indicating whether NA
values should be stripped before the computation proceeds.
The optional column identifier for case weights.
This should be an unquoted column name that evaluates to a numeric column
in data
. For _vec()
functions, a numeric vector.
A single string. Either "first"
or "second"
to specify
which level of truth
to consider as the "event". This argument is only
applicable when estimator = "binary"
. The default uses an
internal helper that generally defaults to "first"
, however, if the
deprecated global option yardstick.event_first
is set, that will be
used instead with a warning.
There is no common convention on which factor level should
automatically be considered the "event" or "positive" result
when computing binary classification metrics. In yardstick
, the default
is to use the first level. To alter this, change the argument
event_level
to "second"
to consider the last level of the factor the
level of interest. For multiclass extensions involving one-vs-all
comparisons (such as macro averaging), this option is ignored and
the "one" level is always the relevant result.
Macro, micro, and macro-weighted averaging is available for this metric.
The default is to select macro averaging if a truth
factor with more
than 2 levels is provided. Otherwise, a standard binary calculation is done.
See vignette("multiclass", "yardstick")
for more information.
Suppose a 2x2 table with notation:
Reference | ||
Predicted | Positive | Negative |
Positive | A | B |
Negative | C | D |
The formulas used here are:
$$Sensitivity = A/(A+C)$$ $$Specificity = D/(B+D)$$ $$Prevalence = (A+C)/(A+B+C+D)$$ $$PPV = (Sensitivity * Prevalence) / ((Sensitivity * Prevalence) + ((1-Specificity) * (1-Prevalence)))$$ $$NPV = (Specificity * (1-Prevalence)) / (((1-Sensitivity) * Prevalence) + ((Specificity) * (1-Prevalence)))$$
See the references for discussions of the statistics.
Max Kuhn
The sensitivity (sens()
) is defined as the proportion of positive
results out of the number of samples which were actually
positive.
When the denominator of the calculation is 0
, sensitivity is undefined.
This happens when both # true_positive = 0
and # false_negative = 0
are true, which mean that there were no true events. When computing binary
sensitivity, a NA
value will be returned with a warning. When computing
multiclass sensitivity, the individual NA
values will be removed, and the
computation will procede, with a warning.
Altman, D.G., Bland, J.M. (1994) ``Diagnostic tests 1: sensitivity and specificity,'' British Medical Journal, vol 308, 1552.
# Two class
data("two_class_example")
sens(two_class_example, truth, predicted)
# Multiclass
library(dplyr)
data(hpc_cv)
hpc_cv %>%
filter(Resample == "Fold01") %>%
sens(obs, pred)
# Groups are respected
hpc_cv %>%
group_by(Resample) %>%
sens(obs, pred)
# Weighted macro averaging
hpc_cv %>%
group_by(Resample) %>%
sens(obs, pred, estimator = "macro_weighted")
# Vector version
sens_vec(
two_class_example$truth,
two_class_example$predicted
)
# Making Class2 the "relevant" level
sens_vec(
two_class_example$truth,
two_class_example$predicted,
event_level = "second"
)
Run the code above in your browser using DataLab