基础模型

基础模型#

此模块为 Mambular 模型提供基类。

模块

描述

BaseModel

使用给定的超参数初始化 BaseModel

TaskModel

用于训练和评估模型的 PyTorch Lightning 模块

Mambular

利用 Mamba 架构和各种归一化技术的 PyTorch 模型

MLP

使用给定的配置初始化MLP模型

ResNet

结构化数据的 ResNet 模型

FTTransformer

利用Transformer架构和各种归一化技术的PyTorch模型

TabTransformer

利用Transformer架构和各种归一化技术的PyTorch模型