concepts.benchmark.logic_induction.graph_dataset.GraphDatasetBase#

class GraphDatasetBase[source]#

Bases: Dataset

Methods

register_datapipe_as_function(function_name, ...)

register_function(function_name, function)

Attributes

functions

__add__(other)#
Parameters:

other (Dataset[T_co]) –

Return type:

ConcatDataset[T_co]

__getitem__(index)#
Return type:

T_co

__init__(nr_nodes, p, epoch_size, directed=False, gen_method='dnc')[source]#
__len__()[source]#
__new__(**kwargs)#
classmethod register_datapipe_as_function(function_name, cls_to_register, enable_df_api_tracing=False)#
classmethod register_function(function_name, function)#
functions: Dict[str, Callable] = {'concat': functools.partial(<function Dataset.register_datapipe_as_function.<locals>.class_function>, <class 'torch.utils.data.datapipes.map.combining.ConcaterMapDataPipe'>, False), 'map': functools.partial(<function Dataset.register_datapipe_as_function.<locals>.class_function>, <class 'torch.utils.data.datapipes.map.callable.MapperMapDataPipe'>, False)}#