hoops_ai.dataset

Dataset Management Module

The Dataset module provides comprehensive tools for managing large-scale CAD datasets efficiently. It handles the complexities of working with collections of CAD models, from exploration and loading to preprocessing and ML integration.

This module is designed to work seamlessly with various dataset formats and provides efficient data loading patterns for machine learning workflows. It supports both traditional ML frameworks and specialized graph-based approaches for CAD data analysis.

For comprehensive examples and dataset management patterns, see the Dataset Management Programming Guide.

Modules

dataset_explorer

dataset_loader

graph_dataset

torch_adapter