sketchgraphs_models.autoconstraint.eval.AutoConstraintPrediction

class sketchgraphs_models.autoconstraint.eval.AutoConstraintPrediction(model, node_feature_mappings, batch_size=32, seed=2, device=None, max_edges=100, mask_function=None)

__init__(model, node_feature_mappings, batch_size=32, seed=2, device=None, max_edges=100, mask_function=None)

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