HGCN DataModule API¶
Config
dataclass
¶
Configuration for training data loader.
HGCNDataModule
¶
Bases: LightningDataModule
Lightning DataModule that materializes triplets once and reuses cached data.
TripletDataset
¶
Bases: Dataset
Dataset for loading triplets.
collate_fn(batch)
¶
Collate function for batching.
create_dataloader(cfg)
¶
Create a PyTorch DataLoader for graph training using the shared streaming cache.