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
)