confusion_matrix: Takes in a trained model with X and y values to produce a...

In UBC-MDS/RMLViz: Machine learning results visualization helper in R

Description

Implementation based on https://stackoverflow.com/a/59119854

Usage

confusion_matrix(actual_y, predicted_y, labels = NULL, title = NULL)

Arguments

actual_y	a data frame of the validation labels
predicted_y	a data frame of predicted values with model
labels	a list of string of confusion matrix labels
title	a string of the confusion matrix label

Value

A plot and a dataframe with scoring metrics

Examples

## Not run: 
library(RMLViz)
data(iris)

set.seed(123)

split <- caTools::sample.split(iris$Species, SplitRatio = 0.75)
training_set <- subset(iris, split == TRUE)
valid_set <- subset(iris, split == FALSE)
X_train <- training_set[, -5]
y_train <- training_set[, 5]
X_valid <- valid_set[, -5]
y_valid <- valid_set[, 5]

predict <- class::knn(X_train, X_train, y_train, k = 5)
confusion_matrix(y_train, predict)

## End(Not run)

UBC-MDS/RMLViz documentation built on April 1, 2020, 4:35 a.m.