我想要使用训练好的模型来预测执行预测的时间

Question

我知道timeit可以用来衡量流逝的时间，但我不知道如何在我的代码中实现。例如，我将评估模型性能如下：

scores = model.evaluate(X_test, Y_test, verbose=0) 

如何添加timeit或其他函数来衡量所需的时间？

Answer 1

尝试：

import time 
... 
t = time.time() 
scores = model.evaluate(X_test, Y_test, verbose=0) 
elapsed_time = time.time()-t 

要检查10000次（更多准确性），尝试：

def check(): 
    t = time.time() 
    scores = model.evaluate(X_test, Y_test, verbose=0) 
    elapsed_time = time.time()-t 
    return elapsed_time 
average_elapsed_time = [sum(check() for i in range(10000))/10000][0] 

Answer 2

这是一个基本的设置：

import time 
start_time = time.time() 
# code 
print("time - {}".format(time.time()-start_time)) 

您也可以使用Python的函数包装器，并为你的函数打包。例如。

import time 

def getime(func): 
    def func_wrapper(*args, **kwargs): 
     start_time = time.time() 
     func(*args, **kwargs) 
     print("function {} completed in - {} seconds".format(
      func.__name__, 
      time.time()-start_time)) 
    return func_wrapper 

# ------------ test example of wrapper --------- # 
@getime 
def foo(): 
    for i in range(1000): 
     for j in range(2000): 
      pass 

foo() 

输出：

function foo completed in - 0.13300752639770508 seconds