Hugging Face Blog·· 2023-04-14AI 评分13
使用 Transformers 库进行图分类
Graph Classification with Transformers
AI 导读
本文介绍如何使用 Hugging Face Transformers 库中的 Graphormer 模型进行图分类任务。通过加载 OGB/ogbg-molhiv 数据集,应用 Graphormer 默认预处理生成节点度数和最短路径矩阵等特征,并加载预训练权重进行微调。该教程展示了从数据加载、格式定义到模型训练与评估的完整流程。
来源:Hugging Face Blog · huggingface.co