Python基础——timeit模块

一般我们需要一段代码（主要是workhorse，程序的核心负重代码，如bp神经网络的整个SGD随机梯度下降的过程）的运行时间：

import timeit

start_time = timeit.default_timer()

for j in range(epochs):
for i in range(n_train_batches):

end_time = timeit.default_timer()
print('ran for %.2fm'%(end_time-start_time)/60.)
# timeit.default_timer()单位是秒

通过任一python IDE（PyCharm or Eclipse），我们可轻易地查看

timeit.default_timer()

的源码：

if sys.platform == "win32":
# On Windows, the best timer is time.clock()
default_timer = time.clock
else:
# On most other platforms the best timer is time.time()
defalut_time = time.time

显然这一层封装的最终目的是为了系统兼容性。

# On Windows
>>> sys.platform
'Win32'

# On CentOS 6.5
>>> sys.platform
'linux2'