missing probability module

Author: topepo

R/mlp_data.R

@@ -72,6 +72,17 @@ mlp_keras_data <-
     )
   )
 
+
+nnet_softmax <- function(results, object) {
+  if (ncol(results) == 1)
+    results <- cbind(1 - results, results)
+    
+  results <- apply(results, 1, function(x) exp(x)/sum(exp(x)))
+  results <- as_tibble(t(results))
+  names(results) <- paste0(".pred_", object$lvl)
+  results
+}
+
 mlp_nnet_data <-
   list(
     libs = "nnet",
@@ -103,6 +114,17 @@ mlp_nnet_data <-
           type = "class"
         )
     ),
+    classprob = list(
+      pre = NULL,
+      post = nnet_softmax,
+      func = c(fun = "predict"),
+      args =
+        list(
+          object = quote(object$fit),
+          newdata = quote(new_data),
+          type = "raw"
+        )
+    ),
     raw = list(
       pre = NULL,
       func = c(fun = "predict"),
@@ -114,6 +136,7 @@ mlp_nnet_data <-
     )
   )
 
+
 # ------------------------------------------------------------------------------
 
 # keras wrapper for feed-forward nnet