Models

Learn how to build the model module.

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.

Press + to interact
ML pipeline architecture
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.

Press + to interact
Diagram of the Model class
Diagram of the Model class

As we did with datasets, we’ll start with an abstract base class called Model ...