R/parsnip-helpers.R

In kindling: Higher-Level Interface of 'torch' Package to Auto-Train Neural Networks

#' Prepare arguments for kindling models
#' @keywords internal
prepare_kindling_args = function(args) {
    args = lapply(args, function(val) {
        if (rlang::is_quosure(val)) {
            return(rlang::eval_tidy(val))
        }
        if (rlang::is_formula(val) && is.null(rlang::f_lhs(val))) {
            return(eval(rlang::f_rhs(val), envir = rlang::f_env(val)))
        }
        val
    })
    
    args = Filter(Negate(is.null), args)
    
    if ("hidden_neurons" %in% names(args) && is.list(args$hidden_neurons)) {
        if (length(args$hidden_neurons) == 1) {
            args$hidden_neurons = args$hidden_neurons[[1]]
        }
    }
    
    if ("activations" %in% names(args) && is.list(args$activations)) {
        if (length(args$activations) == 1) {
            args$activations = args$activations[[1]]
        }
    }
    
    args
}