torch_geometric.data.get_features_for_triplets_groups

get_features_for_triplets_groups(indexer: LargeGraphIndexer, triplet_groups: Iterable[Iterable[Tuple[str, str, str]]], node_feature_name: str = 'x', edge_feature_name: str = 'edge_attr', pre_transform: Optional[Callable[[Tuple[str, str, str]], Tuple[str, str, str]]] = None, verbose: bool = False) → Iterator[Data][source]

给定一个索引器和一系列三元组（如数据集），从索引中检索每个三元组的指定节点和边特征。

Parameters:

  • indexer (LargeGraphIndexer) – 包含所需特征的索引器

  • triplet_groups (Iterable[KnowledgeGraphLike]) – 用于获取特征的三元组列表的列表

  • node_feature_name (str, optional) – 要获取的节点特征。 默认为“x”。

  • edge_feature_name (str, optional) – 要获取的边特征。 默认为“edge_attr”。

  • pre_transform (Optional[Callable[[TripletLike], TripletLike]]) – 可选的预处理操作，用于处理三元组。 默认为 None。

  • verbose (bool, 可选) – 是否打印进度。默认为 False。

Yields:

Iterator[Data] –

For each triplet group, yield a data object containing

来自索引的唯一图表和特征。

Return type:

Iterator[Data]