RayLGBMForecast

ray LightGBM 预测器

lightgbm.ray.RayLGBMRegressor的封装器，添加了一个model_属性，该属性包含已拟合的提升器，并在预测步骤中发送给工作节点。

import lightgbm as lgb
from lightgbm_ray import RayLGBMRegressor

class RayLGBMForecast(RayLGBMRegressor):
    @property
    def model_(self):
        return self._lgb_ray_to_local(lgb.LGBMRegressor)

Give us a ⭐ on Github