跳到正文
原文
Hugging Face Blog·· 2023-04-14AI 评分31

Hugging Face Transformers 库支持使用 Graphormer 进行图分类

Graph Classification with Transformers

AI 导读

Hugging Face Transformers 库目前仅支持 Microsoft 的 Graphormer 模型进行图分类任务。开发者需安装 transformers >= 4.27.2,加载如 ogbg-molhiv 等数据集并应用默认预处理。该流程通过微调预训练检查点实现二分类、回归或多任务分类。

来源:Hugging Face Blog · huggingface.co