[Python] Pytorch
from torch_geometric.data import InMemoryDataset
주로 그래프 데이터를 다룰 때 사용됩니다. 이 클래스는 데이터셋을 메모리에 한 번에 로드하여, 이후 빠르게 접근할 수 있도록 하는 역할을 합니다.
일반적인 사용 패턴
InMemoryDataset을 상속받아 커스텀 데이터셋 클래스를 정의하고, 데이터를 로드하는 방법을 구현하는 것이 일반적입니다. 이 과정에서 init, download, process 등의 메서드를 오버라이드하여 데이터셋의 로딩, 전처리, 다운로드 등의 기능을 구현할 수 있습니다.
댓글남기기