机器学习分类算法:TabM 算法

| 分类 机器学习  分类算法  逻辑回归  | 标签 AI  机器学习  深度学习  神经网络  PyTorch  回归  分类  离散  MLP  TabPFN  TabM 

参考材料:二分类智能诊断全面对比:TabPFN、TAM、LightGBM等十余种算法一键评测!附完整源码

对应的github 仓库:https://github.com/yandex-research/tabm

TabM 是一种新的、高效的表格数据深度学习模型,它在性能、效率和实用性方面都显著优于现有的表格深度学习模型

TabM 依赖于高效集成的方法,其中一个TabM 模型能够高效地模仿一个MLP 集成体,并为每个输入对象生成多个预测。与传统的深度集成方法相比,在TabM 中,其底层隐含的多个MLP 是同时被训练的,并且(默认情况下)共享大部分参数,这带来了显著更好的性能和效率

首先安装依赖

pip install tabm



如果本篇文章对您有所帮助,您可以通过微信(左)或支付宝(右)对作者进行打赏!


上一篇     下一篇