Как я могу получить возвращаемое значение с помощью модуля Python timeit?

Проблема сводится к тому, что timeit._template_func не возвращает возвращаемое функцией значение:

def _template_func(setup, func):
    """Create a timer function. Used if the "statement" is a callable."""
    def inner(_it, _timer, _func=func):
        setup()
        _t0 = _timer()
        for _i in _it:
            _func()
        _t1 = _timer()
        return _t1 - _t0
    return inner

Мы можем подчинить timeit нашей воле с помощью небольшого исправления:

import timeit
import time

def _template_func(setup, func):
    """Create a timer function. Used if the "statement" is a callable."""
    def inner(_it, _timer, _func=func):
        setup()
        _t0 = _timer()
        for _i in _it:
            retval = _func()
        _t1 = _timer()
        return _t1 - _t0, retval
    return inner

timeit._template_func = _template_func

def foo():
    time.sleep(1)
    return 42

t = timeit.Timer(foo)
print(t.timeit(number=1))

возвращается

(1.0010340213775635, 42)

Первое значение — результат timeit (в секундах), второе значение — возвращаемое значение функции.

Обратите внимание, что приведенный выше обезьяний патч влияет только на поведение timeit при передаче вызываемого timeit.Timer. Если вы передаете строковый оператор, вам придется (аналогично) исправить строку timeit.template.

Для Python 3.5 вы можете переопределить значение timeit.template

timeit.template = """
def inner(_it, _timer{init}):
    {setup}
    _t0 = _timer()
    for _i in _it:
        retval = {stmt}
    _t1 = _timer()
    return _t1 - _t0, retval
"""

ответ unutbu работает для python 3.4, но не 3.5, поскольку функция _template_func, похоже, была удалена в 3.5

Как ни странно, я тоже занимаюсь машинным обучением, и у меня аналогичные требования ;-)

Я решил это следующим образом, написав функцию, которая:

• выполняет вашу функцию
• печатает время выполнения вместе с именем вашей функции
• возвращает результаты

Допустим, вы хотите время:

clf = RandomForest(train_input, train_output)

Затем выполните:

clf = time_fn( RandomForest, train_input, train_output )

Stdout покажет что-то вроде:

mymodule.RandomForest: 0.421609s

Код для time_fn:

import time

def time_fn( fn, *args, **kwargs ):
    start = time.clock()
    results = fn( *args, **kwargs )
    end = time.clock()
    fn_name = fn.__module__ + "." + fn.__name__
    print fn_name + ": " + str(end-start) + "s"
    return results

Если я хорошо это понимаю, после Python 3.5 вы можете определять глобальные переменные в каждом экземпляре Timer без необходимости определять их в своем блоке кода. Я не уверен, что у него будут такие же проблемы с распараллеливанием.

Мой подход будет примерно таким:

clf = ensemble.RandomForestClassifier(n_estimators=10)
myGlobals = globals()
myGlobals.update({'clf'=clf})
t = Timer(stmt='clf.fit(trainX,trainy)', globals=myGlobals)
print(t.timeit(number=1))
print(clf)

По состоянию на 2020 год в ноутбуках ipython или jupyter это

t = %timeit -n1 -r1 -o RandomForest(trainX, trainy)
t.best

Если вы не хотите исправлять timeit, вы можете попробовать использовать глобальный список, как показано ниже. Это также будет работать в python 2.7, у которого нет аргумента globals в timeit():

from timeit import timeit
import time

# Function to time - plaigiarised from answer above :-)
def foo():
    time.sleep(1)
    return 42

result = []
print timeit('result.append(foo())', setup='from __main__ import result, foo', number=1)
print result[0]

напечатает время, а затем результат.

Подход, который я использую, заключается в том, чтобы «добавить» время выполнения к результатам функции времени. Итак, я пишу очень простой декоратор, используя модуль «время»:

def timed(func):
    def func_wrapper(*args, **kwargs):
        import time
        s = time.clock()
        result = func(*args, **kwargs)
        e = time.clock()
        return result + (e-s,)
    return func_wrapper

А затем я использую декоратор для функции, которую хочу запрограммировать.

Для Python 3.X я использую этот подход:

# Redefining default Timer template to make 'timeit' return
#     test's execution timing and the function return value
new_template = """
def inner(_it, _timer{init}):
    {setup}
    _t0 = _timer()
    for _i in _it:
        ret_val = {stmt}
    _t1 = _timer()
    return _t1 - _t0, ret_val
"""
timeit.template = new_template