Source code for recbole.model.exlib_recommender.lightgbm

# -*- coding: utf-8 -*-
# @Time   : 2020/1/17
# @Author : Chen Yang
# @Email  : 254170321@qq.com

r"""
recbole.model.exlib_recommender.lightgbm
##########################################
"""

import lightgbm as lgb
from recbole.utils import ModelType, InputType


[docs]class lightgbm(lgb.Booster): r"""lightgbm is inherited from lgb.Booster """ type = ModelType.DECISIONTREE input_type = InputType.POINTWISE def __init__(self, config, dataset): super(lgb.Booster, self).__init__()
[docs] def to(self, device): return self
[docs] def load_state_dict(self, model_file): r"""Load state dictionary Args: model_file (str): file path of saved model """ self = lgb.Booster(model_file=model_file)
[docs] def load_other_parameter(self, other_parameter): r"""Load other parameters """ pass