plot_confusion: plot_confusion

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

Description

plot_confusion

Usage

plot_confusion(..., display = TRUE, return_ggplot = FALSE, title = "",
  subtitle = "")

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

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")
   
}

kerasformula documentation built on May 2, 2019, 9:44 a.m.