跳到正文
原文
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