最近在使用cs框架跑一些因子参数优化的过程中,发现某些因子计算过程中耗费了大量的时间,经过性能分析发现时间耗费主要在于:

- 第一遍性能分析,发现backtrader初始化的时候需要一定的时间,这个看起来是可以避免的
把一些因子回测使用到的模块剥离到单独的文件夹下面,重新运行之后,发现性能有所提升,初始化时候时间减少了6s左右

- 通过进一步分析发现,这个回测过程中主要占用时间的函数是get_b和get_d这两个计算过程,其实耗费时间最多的就是rolling.apply这个函数
- 所以这个优化的核心就是优化rolling.apply这个函数,更具体一点就是改进d = DECAYLINEAR(c, 11)这个函数的算法。在这个函数中,DECAYLINEAR这个函数是通过:
weights = np
本文介绍了在backtrader-cs框架下,通过numba库加速因子计算的过程。针对性能分析中发现的瓶颈函数get_b和get_d,特别是rolling.apply中的DECAYLINEAR函数,应用numba进行优化,实现了约45倍的性能提升,显著缩短了因子参数优化所需的时间。
订阅专栏 解锁全文
1799

被折叠的 条评论
为什么被折叠?



