verticapy.machine_learning.memmodel.preprocessing.OneHotEncoder.transform¶

OneHotEncoder.transform(X: Annotated[list | ndarray, 'Array Like Structure']) → ndarray¶

Transforms and applies the OneHotEncoder model to the input matrix.

Parameters¶

Import the required module.

from verticapy.machine_learning.memmodel.preprocessing import OneHotEncoder

Let’s create a model.

model_ohe = OneHotEncoder(
    categories = [["male", "female"], [1, 2, 3]],
    drop_first = False,
    column_naming = None,
)

Create a dataset.

data = [["male", 1], ["female", 3]]

Transform the data.

model_ohe.transform(data)
Out[4]: 
array([[1, 0, 1, 0, 0],
       [0, 1, 0, 0, 1]])

Note

Refer to OneHotEncoder for more information about the different methods and usages.