Timeit декоратор для примерки Random Forest - PullRequest
Новая
       17

Timeit декоратор для примерки Random Forest

0 голосов
/ 06 января 2020

Я пытаюсь использовать приведенный ниже декоратор, чтобы измерить время, необходимое для подгонки к моей модели случайного леса: 

from functools import wraps
from time import time
def timeit(f):
    @wraps(f)
    def wrapper(*args, **kwargs):
        start = time()
        result = f(*args, **kwargs)
        end = time()
        print 'Elapsed time: {}'.format(end-start)
        return result
    return wrapper

Однако, когда я запускаю код ниже, я не вижу истекшее время: 

from sklearn.ensemble import RandomForestClassifier
rf = RandomForestClassifier(**_rf_params)
timeit(rf.fit)(train_features, train_labels)

Как я могу использовать этот декоратор, чтобы узнать, сколько времени заняла примерка? (работает на Python 2,7)

...