HomePortfolioPostCompositionPhotos

Posts / 태그

#PyTorch Geometric

1편

AI Project
2026-01-21

지식 그래프 기반 노드 분류: GCN을 활용한 NELL 데이터셋 분석

NELL 지식 그래프 데이터셋에서 GCN(Graph Convolutional Network)을 활용하여 노드 분류를 수행했다. PyTorch Geometric으로 모델을 구현하고, 65,755개 노드와 186개 클래스에 대해 97.86%의 테스트 정확도를 달성했다.