对应的github 仓库:https://github.com/yandex-research/tabm
TabM 是一种新的、高效的表格数据深度学习模型,它在性能、效率和实用性方面都显著优于现有的表格深度学习模型
TabM 依赖于高效集成的方法,其中一个TabM 模型能够高效地模仿一个MLP 集成体,并为每个输入对象生成多个预测。与传统的深度集成方法相比,在TabM 中,其底层隐含的多个MLP 是同时被训练的,并且(默认情况下)共享大部分参数,这带来了显著更好的性能和效率
首先安装依赖
pip install tabm