7. Train a Machine Learning Model For Parts Classification

The notebook 4_train_a_ml_model_to_classify_parts.ipynb covers dataset preparation, model training, and evaluation workflows for CAD parts classification using the Experimental FlowTrainer module.

The code and resources for this tutorial can be found in the HOOPS-AI-Tutorials Github repository.

Work inside the hoops_ai_cpu environment to ensure PyTorch, DGL, and other dependencies match the versions documented in </getting_started/evaluate>.

Hint

From the bundle root run jupyter lab notebooks/4_train_a_ml_model_to_classify_parts.ipynb to reproduce the training steps.

• HOOPS AI: EXPERIMENTAL - Flow Trainer
• Make inference, test your current trained model