ConfusionDF: Confusion Matrix (Data Frame Format)

In MLmetrics: Machine Learning Evaluation Metrics

Description

Compute data frame format confusion matrix for internal usage.

Usage

ConfusionDF(y_pred, y_true)

Arguments

y_pred	Predicted labels vector, as returned by a classifier
y_true	Ground truth (correct) 0-1 labels vector

Value

a data.frame of Confusion Matrix

Examples

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

Example output

Attaching package: 'MLmetrics'

The following object is masked from 'package:base':

    Recall

  y_true y_pred Freq
1      0      0   15
2      1      0    1
3      0      1    3
4      1      1   13

MLmetrics documentation built on May 2, 2019, 11:28 a.m.