Source code for cogdl.wrappers.data_wrapper.node_classification.node_classification_dw
from .. import DataWrapper
from cogdl.data import Graph
[docs]class FullBatchNodeClfDataWrapper(DataWrapper):
def __init__(self, dataset):
super(FullBatchNodeClfDataWrapper, self).__init__(dataset)
self.dataset = dataset