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