chapter eight

8 Model training and validation: Part 1

 

This chapter covers

  • Designing modular training components
  • Capturing metrics and artifacts in tracking frameworks
  • Adding model training and validation components to pipelines
  • Using different methods to access training and evaluation data

Building reliable ML systems requires more than just accurate models—it demands reproducible training processes and robust validation strategies. In this chapter, we’ll build on our data preparation pipeline to create production-ready training and validation components (figure 8.1).

Figure 8.1 The mental map where we’re now focusing on the fourth and fifth step/component of our pipeline—model training (4) and evaluation (5)
A screenshot of a computer

AI-generated content may be incorrect.

Through hands-on examples using You Only Look Once (YOLO) object detection, you’ll learn to develop modular training components that can scale from experimentation to production. We’ll explore the intricacies of model validation in real-world scenarios, see how to effectively capture and track metrics for model improvement, and master techniques for seamlessly integrating these components into ML pipelines.

By using these practices with our ID card detection system and later applying them to a movie recommendation system, you’ll gain practical experience in building robust training workflows that form the backbone of production ML systems.

8.1 Training an object detection model

8.1.1 Training YOLO on a custom dataset

8.1.2 Training the model

8.1.3 Container components for system dependencies

8.1.4 Creating the validation component

8.1.5 Creating the pipeline

8.1.6 Executing the pipeline

8.1.7 Validating model artifacts

Summary