rllm.transforms.table_transforms.StackNumerical¶

class rllm.transforms.table_transforms.StackNumerical(out_dim: int)[source]¶

Bases: ColTransform

The StackNumerical class is designed to transform numerical features in tabular data by stacking them into a specified dimension. This transformation changes the shape of the numerical features from [batch_size, num_cols] to [batch_size, num_cols, out_dim], effectively replicating the values along the new dimension.

Parameters:: out_dim (int) – The output dimensionality to which the numerical features will be stacked.