sketchgraphs_models.autoconstraint.dataset

Dataset for auto-constraint model.

Classes

AutoconstraintDataset(sequences, …[, seed])

Functions

sketchgraphs_models.autoconstraint.dataset.collate(batch)
sketchgraphs_models.autoconstraint.dataset.process_node_and_edge_ops(node_ops, edge_ops_in_graph, num_nodes_in_graph, node_feature_mappings)