PyTorch Geometric (PyG) — Trợ lý mạng nơ-ron đồ thị
Skill hướng dẫn Claude viết code PyTorch Geometric chuẩn cho phân loại node/đồ thị, dự đoán liên kết, đồ thị dị thể và huấn luyện quy mô lớn.
Lập trìnhNâng cao★ 33,030⑂ 3,248Điểm AI 8/10Cập nhật lần cuối: 9 thg 8, 2026
Làm được gì
Cung cấp kiến thức thực chiến về PyTorch Geometric (PyG) 2.7:
- Cấu trúc
Data/HeteroDatavà quy tắcedge_indexdạng[2, num_edges](COO) - Bảng so sánh chọn lớp conv: GCNConv, GATv2Conv, SAGEConv, GINConv, HGTConv...
- Mẫu code cho phân loại node, phân loại đồ thị (DataLoader + global pooling), dự đoán liên kết (RandomLinkSplit, negative sampling)
- Mở rộng cho đồ thị lớn với
NeighborLoader,ClusterLoader, GraphSAINT - Ba cách xây GNN dị thể:
to_hetero(),HeteroConv, toán tử gốc nhưHGTConv - Tự viết lớp mới bằng
MessagePassingvới quy ước hậu tố_i/_j - Giải thích mô hình qua
torch_geometric.explain(GNNExplainer, PGExplainer...) - Danh sách 8 lỗi phổ biến kèm cách sửa
Dành cho ai
- Kỹ sư ML/nghiên cứu viên làm việc với đồ thị: hệ gợi ý, knowledge graph, phân tử, mạng xã hội
- Người cần chuyển từ phân tích NetworkX sang huấn luyện GNN thực sự
- Ai từng bối rối vì API PyG thay đổi giữa các phiên bản
Ví dụ sử dụng
- "Viết pipeline GCN 2 lớp phân loại node trên Cora" → sinh code từ Planetoid tới vòng lặp train/eval theo train_mask.
- "Chuyển bảng CSV user–movie thành HeteroData và làm link prediction" → hướng dẫn mapping index,
T.ToUndirected()vàLinkNeighborLoader. - "Đồ thị của tôi quá lớn cho GPU" → tư vấn
num_neighbors, cắt đúngbatch.batch_sizenode hạt giống.
· · · Hướng dẫn cài đặt · · ·
Cài vào ứng dụng Claude (không cần terminal)
- Tải tệp ZIP bằng nút bên dưới.
- Trong Claude, mở Settings → Capabilities và bật 'Code execution and file creation'. (chỉ một lần)
- Vào Customize → Skills → + → 'Upload a skill' rồi tải tệp ZIP lên.
Cài vào Claude Code
Để Claude làm — dán câu dưới đây vào Claude Code
Hãy cài skill mình tìm thấy trên Claude Skill Mart. Copy thư mục skills/torch-geometric từ repo GitHub K-Dense-AI/scientific-agent-skills vào ~/.claude/skills/torch-geometric/ của mình. Sau khi cài xong, cho mình biết skill này làm được gì trong một câu.
Cài bằng lệnh thủ công
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git /tmp/sas && mkdir -p ~/.claude/skills && cp -r /tmp/sas/skills/torch-geometric ~/.claude/skills/⚠ Đây là skill do người khác tạo. Hãy kiểm tra repo gốc trước khi cài.
- Mở terminal.
- Tải kho skill:
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git - Tạo thư mục skill:
mkdir -p ~/.claude/skills - Sao chép skill:
cp -r scientific-agent-skills/skills/torch-geometric ~/.claude/skills/(nhớ giữ cả thư mục conreferences/). - Cài thư viện:
pip install torchrồipip install torch_geometric(cần Python 3.10+, PyTorch 2.6+). - Khởi động lại Claude Code và thử yêu cầu: "Dùng PyG viết mô hình phân loại đồ thị giúp mình".
Xem mã nguồn trên GitHub ↗Giấy phép: MIT