catto_mean {cattonum}R Documentation

Mean encoding

Description

Mean encoding

Usage

catto_mean(train, ..., response = NULL, test = NULL, verbose = TRUE)

Arguments

train

The training data, in a data.frame or tibble.

...

The columns to be encoded. If none are specified, then all character and factor columns are encoded.

response

The response variable used to calculate means.

test

The test data, in a data.frame or tibble.

verbose

Should informative messages be printed? Defaults to TRUE.

Value

The encoded dataset in a cattonum_df if no test dataset was provided, and the encoded datasets in a cattonum_df2 otherwise.

Examples

catto_mean(iris, response = Sepal.Length)

[Package cattonum version 0.0.5 Index]