torch_geometric.datasets.WebQSPDataset

class WebQSPDataset(root: str, split: str = 'train', force_reload: bool = False, use_pcst: bool = True)[source]

Bases: InMemoryDataset

The WebQuestionsSP dataset of the “The Value of Semantic Parse Labeling for Knowledge Base Question Answering” paper.

Parameters:
  • root (str) – Root directory where the dataset should be saved.

  • split (str, optional) – If "train", loads the training dataset. If "val", loads the validation dataset. If "test", loads the test dataset. (default: "train")

  • force_reload (bool, optional) – Whether to re-process the dataset. (default: False)

  • use_pcst (bool, optional) – Whether to preprocess the dataset’s graph with PCST or return the full graphs. (default: True)