torch_geometric.data .Data(同构图)的用法 https://chrsmrrs.github.io/datasets/docs/datasets/这是TUDdataset的连接通过“/tmp/AIDS”选择想要的数据集。 fromtorch_geometric.datasetsimportTUDatasetdataset=TUDataset(root='/tmp/AIDS',name='AIDS')#下载数据data=dataset#TUDataset数据集data1=dataset[0]#第一...
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▼ # 需...