plot_confusion: plot_confusion

In rdrr1990/kerasformula: A High-Level R Interface for Neural Nets

Description

plot_confusion

Usage

plot_confusion(..., display = TRUE, return_ggplot = FALSE,
  title = "", subtitle = "", position = "identity", alpha = 1)

Arguments

...	kms_fit objects. (For each, object$y_test must be binary or categorical.)
display	Logical: display ggplot comparing confusion matrices? (Default TRUE.)
return_ggplot	Default FALSE (if TRUE, returns the ggplot object for further customization, etc.).
title	ggplot title
subtitle	ggplot subtitle
position	Position adjustment, either as a string, or the result of a call to a position adjustment function
alpha	Transparency of points, between 0 and 1

Value

(optional) ggplot. set return_ggplot=TRUE

Examples

if(is_keras_available()){

   model_tanh <- kms(Species ~ ., iris, 
                     activation = "tanh", Nepochs=5, 
                     units=4, seed=1, verbose=0)
   model_softmax <- kms(Species ~ ., iris, 
                        activation = "softmax", Nepochs=5, 
                        units=4, seed=1, verbose=0)
   model_relu <- kms(Species ~ ., iris, 
                     activation = "relu", Nepochs=5, 
                     units=4, seed=1, verbose=0)
                     
   plot_confusion(model_tanh, model_softmax, model_relu, 
                  title="Species", 
                  subtitle="Activation Function Comparison")
   
}

rdrr1990/kerasformula documentation built on June 6, 2019, 8:02 a.m.