rllm.nn.encoder.TabTransformerPreEncoder

class rllm.nn.encoder.TabTransformerPreEncoder(out_dim: int, metadata: Dict[ColType, List[Dict[str, Any]]])[source]

Bases: TablePreEncoder

The TabTransformerEncoder class is a specialized pre-encoder for the TabTransformer model. It initializes a column-specific encoder dict for categorical and numerical features based on the provided metadata. Specifically, it uses EmbeddingEncoder for categorical features and ReshapeEncoder for numerical features.

Parameters:

  • out_dim (int) – The output dimensionality.

  • metadata (Dict[ColType, List[Dict[str, Any]]]) – Metadata for each column type, specifying the statistics and properties of the columns.