diff --git a/NAMESPACE b/NAMESPACE index 2465d298d..b2bede299 100644 --- a/NAMESPACE +++ b/NAMESPACE @@ -5,6 +5,29 @@ S3method(.censoring_weights_graf,model_fit) S3method(augment,model_fit) S3method(autoplot,glmnet) S3method(autoplot,model_fit) +S3method(check_args,C5_rules) +S3method(check_args,bag_tree) +S3method(check_args,boost_tree) +S3method(check_args,cubist_rules) +S3method(check_args,decision_tree) +S3method(check_args,default) +S3method(check_args,discrim_flexible) +S3method(check_args,discrim_linear) +S3method(check_args,discrim_regularized) +S3method(check_args,linear_reg) +S3method(check_args,logistic_reg) +S3method(check_args,mars) +S3method(check_args,mlp) +S3method(check_args,multinom_reg) +S3method(check_args,nearest_neighbor) +S3method(check_args,pls) +S3method(check_args,poisson_reg) +S3method(check_args,rand_forest) +S3method(check_args,surv_reg) +S3method(check_args,survival_reg) +S3method(check_args,svm_linear) +S3method(check_args,svm_poly) +S3method(check_args,svm_rbf) S3method(extract_fit_engine,model_fit) S3method(extract_parameter_dials,model_spec) S3method(extract_parameter_set_dials,model_spec) @@ -180,6 +203,7 @@ export(bartMachine_interval_calc) export(boost_tree) export(case_weights_allowed) export(cforest_train) +export(check_args) export(check_empty_ellipse) export(check_final_param) export(condense_control) diff --git a/R/bag_tree.R b/R/bag_tree.R index 915a13deb..d36997083 100644 --- a/R/bag_tree.R +++ b/R/bag_tree.R @@ -84,6 +84,7 @@ update.bag_tree <- # ------------------------------------------------------------------------------ +#' @export check_args.bag_tree <- function(object) { if (object$engine == "C5.0" && object$mode == "regression") stop("C5.0 is classification only.", call. = FALSE) diff --git a/R/boost_tree.R b/R/boost_tree.R index f56da2fef..c7227eadc 100644 --- a/R/boost_tree.R +++ b/R/boost_tree.R @@ -163,6 +163,7 @@ translate.boost_tree <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.boost_tree <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/c5_rules.R b/R/c5_rules.R index 87ec7bb83..7e5040f5d 100644 --- a/R/c5_rules.R +++ b/R/c5_rules.R @@ -110,6 +110,7 @@ update.C5_rules <- # make work in different places +#' @export check_args.C5_rules <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/cubist_rules.R b/R/cubist_rules.R index be5c8f783..5273a2665 100644 --- a/R/cubist_rules.R +++ b/R/cubist_rules.R @@ -134,6 +134,7 @@ update.cubist_rules <- # make work in different places +#' @export check_args.cubist_rules <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/decision_tree.R b/R/decision_tree.R index e234e43de..b0e7741c0 100644 --- a/R/decision_tree.R +++ b/R/decision_tree.R @@ -127,6 +127,7 @@ translate.decision_tree <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.decision_tree <- function(object) { if (object$engine == "C5.0" && object$mode == "regression") rlang::abort("C5.0 is classification only.") diff --git a/R/discrim_flexible.R b/R/discrim_flexible.R index 330e6bc25..0f5c0162c 100644 --- a/R/discrim_flexible.R +++ b/R/discrim_flexible.R @@ -84,6 +84,7 @@ update.discrim_flexible <- # ------------------------------------------------------------------------------ +#' @export check_args.discrim_flexible <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/discrim_linear.R b/R/discrim_linear.R index 469b8dc8f..54d618a9c 100644 --- a/R/discrim_linear.R +++ b/R/discrim_linear.R @@ -79,6 +79,7 @@ update.discrim_linear <- # ------------------------------------------------------------------------------ +#' @export check_args.discrim_linear <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/discrim_regularized.R b/R/discrim_regularized.R index 776369627..ddf327d22 100644 --- a/R/discrim_regularized.R +++ b/R/discrim_regularized.R @@ -94,6 +94,7 @@ update.discrim_regularized <- # ------------------------------------------------------------------------------ +#' @export check_args.discrim_regularized <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/linear_reg.R b/R/linear_reg.R index f7432bb95..0d4625223 100644 --- a/R/linear_reg.R +++ b/R/linear_reg.R @@ -105,6 +105,7 @@ update.linear_reg <- # ------------------------------------------------------------------------------ +#' @export check_args.linear_reg <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/logistic_reg.R b/R/logistic_reg.R index 92ca9333a..58b5de93e 100644 --- a/R/logistic_reg.R +++ b/R/logistic_reg.R @@ -134,6 +134,7 @@ update.logistic_reg <- # ------------------------------------------------------------------------------ +#' @export check_args.logistic_reg <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/mars.R b/R/mars.R index 86c5330ef..59e4f0f63 100644 --- a/R/mars.R +++ b/R/mars.R @@ -104,6 +104,7 @@ translate.mars <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.mars <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/misc.R b/R/misc.R index 225fdf915..fa591aecc 100644 --- a/R/misc.R +++ b/R/misc.R @@ -282,10 +282,14 @@ show_fit <- function(model, eng) { # Check non-translated core arguments # Each model has its own definition of this +#' @export +#' @keywords internal +#' @rdname add_on_exports check_args <- function(object) { UseMethod("check_args") } +#' @export check_args.default <- function(object) { invisible(object) } diff --git a/R/mlp.R b/R/mlp.R index 909d982e0..d607bae2a 100644 --- a/R/mlp.R +++ b/R/mlp.R @@ -125,6 +125,7 @@ translate.mlp <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.mlp <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/multinom_reg.R b/R/multinom_reg.R index a4ac39b14..8b0d887e0 100644 --- a/R/multinom_reg.R +++ b/R/multinom_reg.R @@ -99,6 +99,7 @@ update.multinom_reg <- # ------------------------------------------------------------------------------ +#' @export check_args.multinom_reg <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/nearest_neighbor.R b/R/nearest_neighbor.R index 41b738095..0976d2907 100644 --- a/R/nearest_neighbor.R +++ b/R/nearest_neighbor.R @@ -97,6 +97,7 @@ positive_int_scalar <- function(x) { # ------------------------------------------------------------------------------ +#' @export check_args.nearest_neighbor <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/pls.R b/R/pls.R index 491389d01..c9eea560f 100644 --- a/R/pls.R +++ b/R/pls.R @@ -86,6 +86,7 @@ update.pls <- # ------------------------------------------------------------------------------ +#' @export check_args.pls <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/poisson_reg.R b/R/poisson_reg.R index b9e103ddd..e538f29ee 100644 --- a/R/poisson_reg.R +++ b/R/poisson_reg.R @@ -100,6 +100,7 @@ translate.poisson_reg <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.poisson_reg <- function(object) { args <- lapply(object$args, rlang::eval_tidy) diff --git a/R/rand_forest.R b/R/rand_forest.R index 43d1331d8..425b9d9fb 100644 --- a/R/rand_forest.R +++ b/R/rand_forest.R @@ -160,6 +160,7 @@ translate.rand_forest <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.rand_forest <- function(object) { # move translate checks here? invisible(object) diff --git a/R/surv_reg.R b/R/surv_reg.R index f5e72431d..e0d321536 100644 --- a/R/surv_reg.R +++ b/R/surv_reg.R @@ -82,6 +82,7 @@ translate.surv_reg <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.surv_reg <- function(object) { if (object$engine == "flexsurv") { diff --git a/R/survival_reg.R b/R/survival_reg.R index 0ab4cb7a9..6b270d825 100644 --- a/R/survival_reg.R +++ b/R/survival_reg.R @@ -81,7 +81,7 @@ translate.survival_reg <- function(x, engine = x$engine, ...) { x } - +#' @export check_args.survival_reg <- function(object) { if (object$engine == "flexsurv") { diff --git a/R/svm_linear.R b/R/svm_linear.R index 178368a40..b888758f3 100644 --- a/R/svm_linear.R +++ b/R/svm_linear.R @@ -139,6 +139,7 @@ translate.svm_linear <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.svm_linear <- function(object) { invisible(object) } diff --git a/R/svm_poly.R b/R/svm_poly.R index d031a2224..028a09294 100644 --- a/R/svm_poly.R +++ b/R/svm_poly.R @@ -133,6 +133,7 @@ translate.svm_poly <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.svm_poly <- function(object) { invisible(object) } diff --git a/R/svm_rbf.R b/R/svm_rbf.R index 8d2d69f8a..af6bde862 100644 --- a/R/svm_rbf.R +++ b/R/svm_rbf.R @@ -138,6 +138,7 @@ translate.svm_rbf <- function(x, engine = x$engine, ...) { # ------------------------------------------------------------------------------ +#' @export check_args.svm_rbf <- function(object) { invisible(object) } diff --git a/man/add_on_exports.Rd b/man/add_on_exports.Rd index 86e840ebd..8ecfcdad4 100644 --- a/man/add_on_exports.Rd +++ b/man/add_on_exports.Rd @@ -4,6 +4,7 @@ \name{null_value} \alias{null_value} \alias{show_fit} +\alias{check_args} \alias{update_dot_check} \alias{new_model_spec} \alias{check_final_param} @@ -18,6 +19,8 @@ null_value(x) show_fit(model, eng) +check_args(object) + update_dot_check(...) new_model_spec(