torch_geometric.datasets.WebQSPDataset

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

Bases: InMemoryDataset

“The Value of Semantic Parse Labeling for Knowledge Base Question Answering”论文中的WebQuestionsSP数据集。

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, 可选) – 是否使用PCST预处理数据集的图或返回完整的图。(默认: True)