Source code for galax.data.datasets.nodes.planetoid

"""Cora, citeseer, pubmed dataset.

Following dataset loading and preprocessing code from tkipf/gcn
https://github.com/tkipf/gcn/blob/master/gcn/utils.py
"""
import galax

[docs]def cora():
    from dgl.data import CoraGraphDataset
    g = CoraGraphDataset()[0]
    g.ndata['h'] = g.ndata['feat']
    del g.ndata['feat']
    g = galax.from_dgl(g)
    return g

[docs]def citeseer():
    from dgl.data import CiteseerGraphDataset
    g = CiteseerGraphDataset()[0]
    g.ndata['h'] = g.ndata['feat']
    del g.ndata['feat']
    g = galax.from_dgl(g)
    return g

[docs]def pubmed():
    from dgl.data import PubmedGraphDataset
    g = PubmedGraphDataset()[0]
    g.ndata['h'] = g.ndata['feat']
    del g.ndata['feat']
    g = galax.from_dgl(g)
    return g