How to add stopwatch to my machine learning alogorithm?

Question

I am quite new to machine learning.

I have created a stopwatatch method:

def stopwatch(model):
    start_time = time.time()
    model
    end_time = ti