当前位置:  开发笔记 > 编程语言 > 正文

提高sklearn中的随机森林回归器的性能

如何解决《提高sklearn中的随机森林回归器的性能》经验,为你挑选了1个好方法。

有一个优化问题,我必须调用随机森林回归器的预测函数数千次。

from sklearn.ensemble import RandomForestRegressor
rfr = RandomForestRegressor(n_estimators=10)
rfr = rfr.fit(X, Y)
for iteration in range(0, 100000):
    # code that adapts the input data according to fitness of the last output
    output_data = rfr.predict(input_data)
    # code that evaluates the fitness of output data

在这种情况下,有没有办法提高预测功能的速度?可能通过使用Cython?



1> 小智..:

您可以使用SKompiler(https://github.com/konstantint/SKompiler)将其转换为C或C ++代码,然后在此处运行。

from skompiler import skompile
expr = skompile(rfr.predict)
with open("output.cpp", "w") as text_file: print(expr.to('sympy/cxx'), file=text_file)

推荐阅读
贴进你的心聆听你的世界
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有