PyTorch Geometric (PyG) 그래프 신경망 도우미
PyG로 노드·링크·그래프 분류부터 이종 그래프와 대규모 이웃 샘플링까지 올바른 패턴으로 구현하도록 돕는 스킬입니다.
개발·코딩고급★ 33,030⑂ 3,248AI 점수 9/10마지막 업데이트: 2026. 8. 9.
무엇을 해주나
Data/HeteroData구조,edge_index의[2, num_edges]규칙 등 PyG의 기본 자료구조를 정확히 잡아줍니다.- GCN·GAT·GraphSAGE·GIN·HGT 등 컨볼루션 레이어 선택 기준을 표로 제시하고, 태스크별(노드 분류/그래프 분류/링크 예측) 학습 루프 코드를 제공합니다.
NeighborLoader,ClusterLoader등 대규모 그래프 확장 전략과to_hetero(),HeteroConv,HGTConv를 활용한 이종 그래프 모델링을 안내합니다.MessagePassing서브클래싱으로 커스텀 레이어를 만드는 방법과torch_geometric.explain기반 설명가능성 도구도 다룹니다.- 자주 틀리는 8가지 함정(활성화 함수 누락, 지연 초기화, 시드 노드 슬라이싱 등)을 체크리스트로 정리합니다.
이런 분께 추천
- 그래프 데이터로 딥러닝 모델을 만드는 ML 엔지니어·연구자
- 추천 시스템, 지식 그래프, 분자 구조 등 이종 그래프를 다루는 개발자
- PyG를 처음 써서
edge_index형식이나 배치 처리에서 자꾸 막히는 분
활용 예시
- "Cora 데이터셋으로 2층 GCN 노드 분류 학습 코드를 만들어줘" → Planetoid 로딩 + 마스크 기반 학습/평가 루프 생성.
- "user-movie 평점 데이터로 이종 그래프 링크 예측 모델을 짜줘" →
HeteroData구성 +to_hetero()변환 +LinkNeighborLoader패턴 제안. - "노드 200만 개 그래프가 GPU 메모리에 안 들어가" →
NeighborLoader의num_neighbors설정과 시드 노드 슬라이싱 방식 안내.
· · · 설치 가이드 · · ·
설치 없이 지금 한 번 써보기
아래를 Claude에 붙여넣으면 설치하지 않고도 이 스킬을 그대로 씁니다.
이 파일의 지시를 읽고 그대로 따라서 나를 도와줘: https://raw.githubusercontent.com/K-Dense-AI/scientific-agent-skills/HEAD/skills/torch-geometric/SKILL.md 내가 원하는 것: (여기에 하고 싶은 일을 쓰세요)
Claude가 링크를 열지 못하면, 링크를 직접 열어 내용을 복사해 붙여넣으세요.
↓ 쓸 만하면 아래에서 ZIP을 받아 설치하세요. 그러면 매번 붙여넣지 않아도 알아서 작동합니다.
Claude 앱에 설치 (터미널 필요 없음)
- 아래 버튼으로 ZIP 파일을 받으세요.
- Claude 설정 → Capabilities에서 '코드 실행 및 파일 생성'을 켭니다. (한 번만)
- Claude에서 Customize → Skills → + → '스킬 업로드'를 누르고 받은 ZIP을 올립니다.
Claude Code에 설치
Claude에게 맡기기 — 아래 문장을 Claude Code에 붙여넣으세요
클로드스킬마트에서 찾은 스킬을 설치해줘. GitHub 저장소 K-Dense-AI/scientific-agent-skills 의 skills/torch-geometric 폴더를 내 ~/.claude/skills/torch-geometric/ 에 그대로 복사해줘. 설치가 끝나면 이 스킬로 무엇을 할 수 있는지 한 줄로 알려줘.
직접 명령으로 설치하기
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git && mkdir -p ~/.claude/skills && cp -r scientific-agent-skills/skills/torch-geometric ~/.claude/skills/⚠ 제3자가 만든 스킬입니다. 설치 전 원본 저장소를 한 번 확인하세요.
- 터미널을 열고 스킬 저장소를 내려받습니다:
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git - 스킬 폴더가 없다면 만듭니다:
mkdir -p ~/.claude/skills - 이 스킬만 복사합니다:
cp -r scientific-agent-skills/skills/torch-geometric ~/.claude/skills/ references/하위 마크다운 문서들이 함께 복사되었는지 확인합니다.- 실행 환경에 라이브러리를 설치합니다:
pip install torch후pip install torch_geometric(Python 3.10+, PyTorch 2.6+ 권장). - Claude Code를 다시 실행하고 "PyG로 그래프 분류 모델 만들어줘"처럼 요청하면 스킬이 자동 발동합니다.
GitHub에서 원본 보기 ↗라이선스: MIT