Check Classification Models

This function verifies if the given machine learning models support classification tasks by ensuring the forClass column is TRUE in their model lookup details.

Usage

check_forclass(models)

Arguments

models

A character vector of model names to check.

Value

A data frame containing model details where forClass is TRUE.

Examples

check_forclass(c("lvq", "gbm", "svmRadial"))
#>       model         parameter                   label forReg forClass probModel
#> 1       lvq              size           Codebook Size  FALSE     TRUE     FALSE
#> 2       lvq                 k             #Prototypes  FALSE     TRUE     FALSE
#> 3       gbm           n.trees   # Boosting Iterations   TRUE     TRUE      TRUE
#> 4       gbm interaction.depth          Max Tree Depth   TRUE     TRUE      TRUE
#> 5       gbm         shrinkage               Shrinkage   TRUE     TRUE      TRUE
#> 6       gbm    n.minobsinnode Min. Terminal Node Size   TRUE     TRUE      TRUE
#> 7 svmRadial             sigma                   Sigma   TRUE     TRUE      TRUE
#> 8 svmRadial                 C                    Cost   TRUE     TRUE      TRUE