rllm.transforms.table_transforms.TabTransformerTransform¶

class rllm.transforms.table_transforms.TabTransformerTransform(out_dim: int, metadata: Dict[ColType, List[Dict[str, Any]]] | None = None)[source]¶

Bases: TableTransform

TabTransformerTransform applies ColNormalize transform to tabular data specifically for the TabTransformer model.

Parameters:

out_dim (int) – The output dimensionality.
metadata (Dict[ColType, List[Dict[str, Any]]], optional) – Metadata containing information about the columns, such as statistics. (default: None)

reset_parameters() → None[source]¶: Initialize the parameters of post_module.