Models
Explore the development of a model module within a machine learning pipeline. Understand abstract base classes, inheritance, and mixins for training, evaluation, and reporting. Gain hands-on experience modifying and training models using scikit-learn.
We'll cover the following...
We'll cover the following...
The Model module
Now that we’ve built the dataset module, let’s work on the model module. The model module implements the model operations shown in the architecture diagram below.
ML pipeline architecture
The module needs to be able to load a model from disk, train it, evaluate its performance, and save it to disk, as shown in the diagram below.
Diagram of the Model class
As we did with datasets, we’ll start with an abstract base class called Model ...