torch_frame.nn.encoder.MultiCategoricalEmbeddingEncoder

class MultiCategoricalEmbeddingEncoder(out_channels: int | None = None, stats_list: list[dict[StatType, Any]] | None = None, stype: stype | None = None, post_module: torch.nn.Module | None = None, na_strategy: NAStrategy | None = None, mode: str = 'mean')

基础类：StypeEncoder

一个基于嵌入查找的多分类特征编码器。它为每个分类特征应用torch.nn.EmbeddingBag并连接输出嵌入。

Parameters:

mode (str) – “sum”, “mean” 或 “max”。 指定减少包的方式。（默认：mean）

reset_parameters() → None

初始化post_module的参数。

encode_forward(feat: MultiNestedTensor, col_names: list[str] | None = None) → Tensor

主要的前向函数。将输入 feat 从 TensorFrame（形状 [batch_size, num_cols]）映射到输出 x，其形状为 [batch_size, num_cols, out_channels]。