Going Classy
Learn how you can build a class to handle the model training part.
We'll cover the following
Building classes for model training
So far, the %%writefile
magic has helped us to organize the code into three distinct parts: data preparation, model configuration, and model training. At the end of the chapter, Rethinking the Training Loop, though, we bumped into some of its limitations, e.g., being unable to choose a different number of epochs without editing the model training code.
Clearly, this situation is not ideal. We need to do better. We need to go classy; that is, we need to build a class to handle the model training part.
The class
Let us start defining our class with a rather unoriginal name: StepByStep
. Since we are starting it from scratch, either we do not specify a parent class or we inherit it from the fundamental object
class. I prefer the latter, so our class definition looks like this:
Get hands-on with 1400+ tech skills courses.