Machine Learning - Tools/PyG
PyG 시작하기 (PyG tutorial, PyG 사용법)
이번 글에서는 PyG를 이용해서 GNN 모델링을 하려면 어떤 흐름으로 모델링을 진행해야 하는지에 대해 간략하게 알아볼 것이다. PyG의 설치 방법은 다음 글을 참고하면 된다. * 이 글은 pyG 2.1.0 버전을 기준으로 작성되었습니다. 이 글은 파이썬에 대한 기본 지식 및 GNN의 기본 용어에 대한 이해가 있다는 가정 하에 진행됩니다. Library / module import import os import torch import networkx as nx import matplotlib.pyplot as plt from torch.nn import Linear from torch_geometric.nn import GCNConv from torch_geometric.datasets import Kar..
2022. 8. 25. 19:23
최근댓글