sketchgraphs_models.graph.dataset.GraphDataset

class sketchgraphs_models.graph.dataset.GraphDataset(sequences, node_feature_mapping=None, edge_feature_mapping=None, seed=None)

Dataset for numerical constraint model.

This dataset processes the main data format for graph models, that is, it represents (partial) sketches as graphs which are suitable for computation using graph neural networks. Additionally, it also optionally outputs sparse features for the entities or the edges.

__init__(sequences, node_feature_mapping=None, edge_feature_mapping=None, seed=None)

Initialize self. See help(type(self)) for accurate signature.