Hugging Face Blog·· 2023-04-14AI 評分53
使用 Hugging Face Transformers 進行圖分類
Graph Classification with Transformers
AI 導讀
Hugging Face Blog 示範如何使用 Transformers 中的 Graphormer 模型進行圖分類,涵蓋資料載入、預處理及微調訓練。示例採用 OGB/ogbg-molhiv 資料集,載入 clefourrier/pcqm4mv2_graphormer_base,將任務設為二分類後以 Trainer 訓練。
來源:Hugging Face Blog · huggingface.co