added an extra check on column names

topepo · topepo · commit 1fc7dfeb9489 · 2019-07-06T18:51:14.000-04:00
diff --git a/R/mlp.R b/R/mlp.R
@@ -375,7 +375,7 @@ keras_mlp <-
   }
 
 keras_numeric_post <- function(results, object) {
-  if (ncol(results) > 1) {
+  if (ncol(results) > 1 && !is.null(object$fit$y_names)) {
     colnames(results) <- object$fit$y_names
   }
   maybe_multivariate(results, object)

Original file line number	Diff line number	Diff line change
`@@ -375,7 +375,7 @@ keras_mlp <-`
`375`	`375`	`}`
`376`	`376`
`377`	`377`	`keras_numeric_post <- function(results, object) {`
`378`		`- if (ncol(results) > 1) {`
	`378`	`+ if (ncol(results) > 1 && !is.null(object$fit$y_names)) {`
`379`	`379`	`colnames(results) <- object$fit$y_names`
`380`	`380`	`}`
`381`	`381`	`maybe_multivariate(results, object)`