tab-transformer-pytorch

收录理由

表格数据建模目前大多还是梯度提升的天下，但这个仓库给想试试注意力机制的人提供了一个轻量入口——不必推翻现有技术栈，就能在同一类数据上验证效果。它实现了2020年论文中的原始TabTransformer，作者称其性能已逼近GBDT，同时还打包了Yandex的FT-Transformer，后者用更简单的方案嵌入连续数值。两个模型装在一个包里，方便你直接和自己的管线做对比实验。代码就是个朴素的PyTorch模块，体积小、能pip安装，分类或回归任务都能直接用，不需要引入额外的大框架。如果你正纠结结构化数据上深度学习能否胜过手头方案，拿自己的数据跑一跑，成本很低。