Transformers로 그래프 분류하기

Hugging Face의 Transformers 라이브러리를 사용하여 그래프 분류(Graph Classification)를 수행하는 기술적 방법을 다룬다. 현재 라이브러리에서 지원하는 그래프 트랜스포머 모델인 Microsoft의 Graphormer를 중심으로 설명한다.

데이터 로딩 및 형식 datasets 라이브러리를 통해 OGB(Open Graph Benchmark) 데이터셋을 로드하는 방법과 edge_index, num_nodes, y, node_feat, edge_attr로 구성된 그래프 데이터의 표준 형식을 상세히 안내한다.

전처리(Preprocessing) Graphormer 모델의 학습을 돕기 위해 입/출력 차수(degree) 정보 및 노드 간 최단 경로 행렬 등을 생성하는 전처리 과정을 포함한다.