ReasonGNN/lrgb-pascalvoc-neuralwalker-walk-length-6

import torch from datasets import load_dataset from torch_geometric.data import Data, Batch from tqdm import tqdm # Load the HuggingFace dataset from hub hf_dataset = load_dataset("samm393/lrgb-pascalvoc-neuralwalker") # Load multiple graphs num_graphs = 10 # Convert each to PyG Data object pyg_graphs = [] for i in tqdm(range(num_graphs), desc="Loading graphs"): hf_row = hf_dataset['train'][i] pyg_dict = {k: torch.tensor(v) if isinstance(v, list) else v for k, v in hf_row.items()}… See the full description on the dataset page: https://huggingface.co/datasets/ReasonGNN/lrgb-pascalvoc-neuralwalker-walk-length-6.