select_learn: Select 'rtemis' Learner

View source: R/select_learn.R

select_learnR Documentation

Select rtemis Learner

Description

Accepts learner name (supports abbreviations) and returns rtemis function name or the function itself. If run with no parameters, prints list of available algorithms.

Usage

select_learn(alg, fn = FALSE, name = FALSE, desc = FALSE)

Arguments

alg

Character: Model name. Case insensitive. e.g. "XGB" for xgboost

fn

Logical: If TRUE, return function, otherwise name of function. Defaults to FALSE

name

Logical: If TRUE, return canonical name of algorithm alg

desc

Logical: If TRUE, return full name / description of algorithm alg

Value

function or name of function (see param fn) or short algorithm name (name = TRUE) or full algorithm name (desc = TRUE)

Author(s)

E.D. Gennatas


egenn/rtemis documentation built on April 24, 2024, 6:58 p.m.