Skip to content

GraphGallery 0.1.9

Compare
Choose a tag to compare
@EdisonLeeeee EdisonLeeeee released this 20 Aug 10:00
· 668 commits to master since this release

Changes

  • Add datasets support, Planetoid and SimpleGraphDataset:
from graphgallery.data import SimpleGraphDataset, Planetoid

data =  Planetoid('cora') # cora, citeseer, pubmed
# or 
data = SimpleGraphDataset('cora') # cora, citeseer, pubmed, cora_ml, polblogs

which consists of

adj = data.adj # scipy sprase matrix
x = data.x  # numpy array
labels = data.labels  # numpy array
idx_train = data.idx_train  # numpy array
idx_val = data.idx_val  # numpy array
idx_test = data.idx_test  # numpy array

for more details, please refer to https://github.com/EdisonLeeeee/GraphData