sketchgraphs_models.graph.sample

This module implements sampling from the entity model.

Classes

GraphSamplingModel(model_core, entity_label, …)

Functions

sketchgraphs_models.graph.sample.generate_sample(model: sketchgraphs_models.graph.sample.GraphSamplingModel, max_iters, node_feature_mapping, edge_feature_mapping, generator=None, device=None)

sketchgraphs_models.graph.sample.load_saved_model(model_state_path)

sketchgraphs_models.graph.sample.main()

sketchgraphs_models.graph.sample.sample_and_print(args)

sketchgraphs_models.graph.sample.sample_and_write_pickle(args, output_path, proc_idx)

sketchgraphs_models.graph.sample.sample_and_write_pickle_multithreaded(args)

sketchgraphs_models.graph.sample.sample_iterator(args)