처리의 소요 시간이 명확한 코드는 고맙다.

코드 템플릿

import time
from contextlib import contextmanager
from time import sleep

@contextmanager
def timer(title):
    t0 = time.time()
    yield
    print("{} - done in {:.0f}s".format(title, time.time() - t0))

def data_import():
    sleep(10)

def data_preprocessing():
    sleep(20)

def train():
    sleep(30)

def main():
    with timer("Process data_import"):
        data_import()
    with timer("Process data_preprocessing"):
        data_preprocessing()
    with timer("Process train"):
        train()


if __name__ == "__main__":
    with timer("Full model run"):
        main()

실행 결과

Process data_import - done in 10s
Process data_preprocessing - done in 20s
Process train - done in 30s
Full model run - done in 60s