Skip to content

Commit

Permalink
fix: ties method to auto_fselector
Browse files Browse the repository at this point in the history
  • Loading branch information
be-marc committed Dec 16, 2023
1 parent 4d85646 commit ca61f35
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 3 deletions.
20 changes: 18 additions & 2 deletions R/auto_fselector.R
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,25 @@
#' @template param_store_models
#' @template param_check_values
#' @template param_callbacks
#' @template param_ties_method
#'
#' @export
#' @inherit AutoFSelector examples
auto_fselector = function(fselector, learner, resampling, measure = NULL, term_evals = NULL, term_time = NULL, terminator = NULL, store_fselect_instance = TRUE, store_benchmark_result = TRUE, store_models = FALSE, check_values = FALSE, callbacks = list()) {
auto_fselector = function(
fselector,
learner,
resampling,
measure = NULL,
term_evals = NULL,
term_time = NULL,
terminator = NULL,
store_fselect_instance = TRUE,
store_benchmark_result = TRUE,
store_models = FALSE,
check_values = FALSE,
callbacks = list(),
ties_method = "n_features"
) {
terminator = terminator %??% terminator_selection(term_evals, term_time)

AutoFSelector$new(
Expand All @@ -35,5 +50,6 @@ auto_fselector = function(fselector, learner, resampling, measure = NULL, term_e
store_benchmark_result = store_benchmark_result,
store_models = store_models,
check_values = check_values,
callbacks = callbacks)
callbacks = callbacks,
ties_method = ties_method)
}
12 changes: 11 additions & 1 deletion man/auto_fselector.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit ca61f35

Please sign in to comment.