How Could I Define These Methods within a Class for a Model? (Python)

Question

I have the following class, UNetwork. It is there for reference. The actual question is related to the second class, Model.

class UN