torch_geometric.data.get_features_for_triplets
- get_features_for_triplets(indexer: LargeGraphIndexer, triplets: 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) Data[source]
- For a given set of triplets retrieve a Data object containing the
从索引中获取独特的图形和特征。
- Parameters:
indexer (LargeGraphIndexer) – Indexer containing desired features
triplets (KnowledgeGraphLike) – 用于获取特征的三元组
node_feature_name (str, optional) – 用于节点特征的特征。 默认为“x”。
edge_feature_name (str, optional) – 用于边缘特征的特征。 默认为“edge_attr”。
pre_transform (可选[Callable[[TripletLike], TripletLike]]) – 可选的三元组预处理函数。默认为 None。
verbose (bool, optional) – Whether to print progress. Defaults to False.
- Returns:
- 包含唯一图和特征的数据对象
给定三元组的索引。
- Return type: