# -*- coding: utf-8 -*-
# @Time : 2020/11/19
# @Author : Chen Yang
# @Email : 254170321@qq.com
r"""
recbole.model.exlib_recommender.xgboost
########################################
"""
import xgboost as xgb
from recbole.utils import ModelType, InputType
[docs]class XGBoost(xgb.Booster):
r"""XGBoost is inherited from xgb.Booster"""
type = ModelType.DECISIONTREE
input_type = InputType.POINTWISE
def __init__(self, config, dataset):
super().__init__(params=None, cache=(), model_file=None)
[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.load_model(model_file)
[docs] def load_other_parameter(self, other_parameter):
r"""Load other parameters"""
pass