官方仓库:https://github.com/pyg-team/pytorch_geometric 官方文档:PyG Documentationtorch_geometric.data .Data(同构图)的用法labelnamedescribeshape√data.keys返回属性名列表√data.y真实标签√data.x节点…
out = subgraph(subset, edge_index, edge_attr, relabel_nodes=True)assertout[0].tolist() == [[0,1,1,2], [1,0,2,1]]assertout[1].tolist() == [7,8,9,10] 开发者ID:rusty1s,项目名称:pytorch_geometric,代码行数:21,代码来源:test_subgraph.py 示例8: test_batched_negative_sampling ...
explainer.visualize_subgraph(2, edge_index, edge_mask, y=y, threshold=None) explainer.visualize_subgraph(2, edge_index, edge_mask, y=y, threshold=0.5) 开发者ID:rusty1s,项目名称:pytorch_geometric,代码行数:22,代码来源:test_gnn_explainer.py 示例7: test_deep_graph_infomax ▲点赞 6▼ # 需...