文章介绍NVIDIA新推出的Structured Data Models库,该库将TabICLv2、KumoTabular和KumoRelational等表格预训练基础模型统一到单一PyTorch接口下,支持无需训练的直接预测。其核心机制是上下文学习:用户提供带标签的示例行和待预测行,模型在一次前向传播中即返回概率或完整预测分布。教程演示了与scikit-learn训练基线的对比、跨多次查询复用编码上下文、从回归任务读取校准预测区间,以及扩展到二十类分类。文章还指出默认配置的隐患,如预处理会丢弃时间戳和文本列、解释器存在两处陷阱,并构建三表商店数据库演示关系型预测,强调区分可靠关系模型与数据泄漏模型的关键参数。该工作降低了表格数据建模门槛,对快速部署和数据库级预测具有实用意义。
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅