recbole.model.exlib_recommender.xgboost

class recbole.model.exlib_recommender.xgboost.XGBoost(config, dataset)[source]

Bases: xgboost.core.Booster

XGBoost is inherited from xgb.Booster

input_type = 1
load_other_parameter(other_parameter)[source]

Load other parameters

load_state_dict(model_file)[source]

Load state dictionary

Parameters

model_file (str) – file path of saved model

to(device)[source]
type = 6