reshape_task_embedding: Reshape a Task for use with entity embeddings.

In mlr-org/mlr3keras: mlr3 Keras extension

Reshape a Task for use with entity embeddings.

Description

• logical variables are treated as integers and converted to either 0 or 1.

• continuous variables are stored in a matrix "continuous"

• categorical variables are integer encoded and stored as a single list element each.

Usage

reshape_task_embedding(task)

Arguments

task	Task A mlr3 Task.

Value

A list with slots data:the reshaped data and fct_levels: the levels corresponding to each factor feature.

Examples

task = mlr3::mlr_tasks$get("boston_housing")
reshape_task_embedding(task)

mlr-org/mlr3keras documentation built on April 12, 2022, 11:35 a.m.