torch_frame.nn.encoder.LinearPeriodicEncoder
- class LinearPeriodicEncoder(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, n_bins: int | None = 16)[source]
基础类:
StypeEncoder一个周期性编码器,利用正弦函数将输入张量转换为三维张量。编码使用可训练的参数定义,并包括正弦和余弦函数的应用。原始编码在“On Embeddings for Numerical Features in Tabular Deep Learning”中描述。
- Parameters:
n_bins (int) – 用于周期性编码的箱数。