최대 1 분 소요

from torch_geometric.data import InMemoryDataset

주로 그래프 데이터를 다룰 때 사용됩니다. 이 클래스는 데이터셋을 메모리에 한 번에 로드하여, 이후 빠르게 접근할 수 있도록 하는 역할을 합니다.

일반적인 사용 패턴

InMemoryDataset을 상속받아 커스텀 데이터셋 클래스를 정의하고, 데이터를 로드하는 방법을 구현하는 것이 일반적입니다. 이 과정에서 init, download, process 등의 메서드를 오버라이드하여 데이터셋의 로딩, 전처리, 다운로드 등의 기능을 구현할 수 있습니다.

댓글남기기