HyperAIHyperAI

Command Palette

Search for a command to run...

PyTorch Frame:一种用于多模态表格学习的模块化框架

Weihua Hu Yiwen Yuan Zecheng Zhang Akihiro Nitta Kaidi Cao Vid Kocijan Jinu Sunil Jure Leskovec Matthias Fey

摘要

我们介绍了PyTorch Frame,这是一个基于PyTorch的框架,用于处理多模态表格数据的深度学习。PyTorch Frame通过提供一个基于PyTorch的数据结构来处理复杂的表格数据,引入模型抽象以实现表格模型的模块化实施,并允许外部基础模型(例如,大型语言模型(LLMs)用于文本列)的集成,从而简化了表格深度学习的过程。我们通过以模块化方式实现多种表格模型,成功将其应用于复杂的多模态表格数据,并将我们的框架与PyTorch Geometric(一个用于图神经网络(GNNs)的PyTorch库)集成,实现了在关系数据库上的端到端学习,展示了PyTorch Frame的实用性。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供