sketchgraphsΒΆ

Tools for manipulating and learning with the SketchGraphs dataset.

The sub-modules of this module implement the required operations to create datasets at various levels of abstraction from the source data. In particular, there are three main scripts used to produce training data for the models.

Modules

sketchgraphs.data

This module contains the main data representations for the SketchGraphs dataset.

sketchgraphs.onshape

This module contains basic code to interact with the Onshape API.

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.