Coverage for hopwise/model/exlib_recommender/lightgbm.py: 85%

1# @Time : 2020/1/17 

2# @Author : Chen Yang 

3# @Email : 254170321@qq.com 

4 

5r"""hopwise.model.exlib_recommender.lightgbm 

6########################################## 

7""" 

8 

9import lightgbm as lgb 

10 

11from hopwise.utils import InputType, ModelType 

12 

13 

14class LightGBM(lgb.Booster): 

15 r"""LightGBM is inherited from lgb.Booster""" 

16 

17 type = ModelType.DECISIONTREE 

18 input_type = InputType.POINTWISE 

19 

20 def __init__(self, config, dataset): 

21 super(lgb.Booster, self).__init__() 

22 

23 def to(self, device): 

24 return self 

25 

26 def load_state_dict(self, model_file): 

27 r"""Load state dictionary 

28 

29 Args: 

30 model_file (str): file path of saved model 

31 

32 """ 

33 self = lgb.Booster(model_file=model_file) # noqa: F841, PLW0642 

34 

35 def load_other_parameter(self, other_parameter): 

36 r"""Load other parameters""" 

37 pass