본문 바로가기

Programming/Python

PyG 란?

728x90
반응형
 

PyG Documentation — pytorch_geometric 2.0.2 documentation

PyG Documentation In addition, it consists of easy-to-use mini-batch loaders for operating on many small and single giant graphs, multi GPU-support, a large number of common benchmark datasets (based on simple interfaces to create your own), and helpful tr

pytorch-geometric.readthedocs.io

 

Pytorch는 딥러닝을 공부하는 분들이라면 다들 알고 있겠지만 PyG(pytorch_geometric)은 생소한 사람들도 많을 것 같다.

PyG는 pytorch를 기반으로 쉽게 Graph Neural Networks를 구현할 수 있도록 도와주는 라이브러리이다.

나같은 경우는 추천 시스템의 다양한 논문들을 읽어보면서 그래프를 사용한 추천 시스템에 관심이 생겨서 알게 되었다.

 

다음에 기회가 된다면 간단한 튜토리얼을 한 번 올려볼까 한다.

참고로 PyG를 사용해서 구현한 논문의 코드를 돌려봤는데 모든 라이브러리가 그렇긴 하지만 버전 차이가 매우 중요한 것 같으니 사용할 때 주의하는게 좋을 듯하다.

728x90
반응형