Pass additional parameters to keras::evaluate

For example, verbosity can be reduced with `verbose = 0`. Refs #27
fdavidcl · Jan 12, 2019 · 14c4fd0 · 14c4fd0
1 parent 945e50a
commit 14c4fd0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/evaluate.R b/R/evaluate.R
@@ -10,7 +10,7 @@
 #'   just like the ones defined in `\link[ruta]{evaluate}`.
 #' @seealso `\link[ruta]{evaluate}`
 #' @export
-evaluation_metric <- function(evaluate_f) function(learner, data) {
+evaluation_metric <- function(evaluate_f) function(learner, data, ...) {
   k_model <- learner$models$autoencoder
 
   keras::compile(
@@ -19,7 +19,7 @@ evaluation_metric <- function(evaluate_f) function(learner, data) {
     loss = to_keras(learner$loss, learner),
     metrics = evaluate_f
   )
-  keras::evaluate(k_model, x = data, y = data)
+  keras::evaluate(k_model, x = data, y = data, ...)
 }
 
 #' Evaluation metrics