LightGBM 尝试通过多线程来加速许多操作。这些操作中使用的线程数可以通过传递给 lgb.train 和 lgb.Dataset 等函数的 params
中的 num_threads
参数来控制。然而,一些操作(例如从文本文件具体化模型)是通过不明确接受线程控制配置的代码路径完成的。
使用此函数可以设置 LightGBM 对此类操作使用的最大线程数。
此函数会影响同一进程中的所有 LightGBM 操作。
因此,例如,如果您调用 setLGBMthreads(4)
,同一进程中的其他多线程 LightGBM 操作将不会使用超过 4 个线程。
调用 setLGBMthreads(-1)
可以移除此限制。
setLGBMthreads(num_threads)