###########################################################
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 <https://github.com/techsoft3d/HOOPS-AI-tutorials/tree/main>`__ Github repository.

Work inside the ``hoops_ai_cpu`` environment to ensure PyTorch, DGL, and other dependencies match the versions documented in :doc:`</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.

.. toctree::
   :maxdepth: 1
   :titlesonly:

   /tutorials/hoops_ai_tutorials/notebooks/4_train_a_ml_model_to_classify_parts.ipynb