LightGBM 尝试通过多线程来加速许多操作。这些操作中使用的线程数可以通过传递给 lgb.trainlgb.Dataset 等函数的 params 中的 num_threads 参数来控制。然而,一些操作(例如从文本文件具体化模型)是通过不明确接受线程控制配置的代码路径完成的。

使用此函数可以设置 LightGBM 对此类操作使用的最大线程数。

此函数会影响同一进程中的所有 LightGBM 操作。

因此,例如,如果您调用 setLGBMthreads(4),同一进程中的其他多线程 LightGBM 操作将不会使用超过 4 个线程。

调用 setLGBMthreads(-1) 可以移除此限制。

setLGBMthreads(num_threads)

参数

num_threads

LightGBM 在多线程操作中使用的最大线程数

另请参阅