sketchgraphs.pipeline

This module implements the transformations in our data pipeline, taking the raw JSON data obtained from Onshape into data that is suitable for model training.

Modules

sketchgraphs.pipeline.graph_model	This module provides the data pipeline to convert from the stored sequence format to a format suitable for training our graph-based deep learning model.
sketchgraphs.pipeline.make_quantization_statistics	This script computes statistics required for quantization of continuous parameters.
sketchgraphs.pipeline.numerical_parameters	This module implements functionality to handle numerical parameters in the sketch.