Get Started
User Guide
Developer Guide
API REFERENCE:
Config
AFM
AutoInt
DCN
DeepFM
DSSM
FFM
FieldAwareFactorizationMachine
FM
FNN
FwFM
LR
NFM
InnerProductLayer
OuterProductLayer
PNN
WideDeep
xDeepFM
AbstractRecommender
AutoEncoderMixin
ContextRecommender
GeneralRecommender
KnowledgeRecommender
SequentialRecommender
xavier_normal_initialization()
xavier_uniform_initialization()
AttLayer
BaseFactorizationMachine
BiGNNLayer
CNNLayers
ContextSeqEmbAbstractLayer
ContextSeqEmbLayer
Dice
FLEmbedding
FMEmbedding
FMFirstOrderLinear
FeatureSeqEmbLayer
FeedForward
ItemToInterestAggregation
LightMultiHeadAttention
LightTransformerEncoder
LightTransformerLayer
MLPLayers
MultiHeadAttention
SequenceAttLayer
SparseDropout
TransformerEncoder
TransformerLayer
VanillaAttention
activation_layer()
BPRLoss
EmbLoss
EmbMarginLoss
RegLoss
load_data_and_model()
objective_function()
run_recbole()
run_recboles()
AbstractSampler
KGSampler
RepeatableSampler
Sampler
SeqSampler
ExhaustiveSearchError
HyperTuning
exhaustive_search()
AbstractTrainer
DecisionTreeTrainer
KGATTrainer
KGTrainer
LightGBMTrainer
MKRTrainer
NCLTrainer
PretrainTrainer
RaCTTrainer
RecVAETrainer
S3RecTrainer
TraditionalTrainer
Trainer
XGBoostTrainer
full_sort_scores()
full_sort_topk()
EvaluatorType
FeatureSource
FeatureType
InputType
KGDataLoaderState
ModelType
RemoveColorFilter
init_logger()
set_color()
calculate_valid_score()
dict2str()
early_stopping()
ensure_dir()
get_flops()
get_gpu_usage()
get_local_time()
get_model()
get_tensorboard()
get_trainer()
init_seed()
AFM.afm_layer()
AFM.build_cross()
AFM.calculate_loss()
AFM.forward()
AFM.predict()
AFM.training
AutoInt.autoint_layer()
AutoInt.calculate_loss()
AutoInt.forward()
AutoInt.predict()
AutoInt.training
DCN.calculate_loss()
DCN.cross_network()
DCN.forward()
DCN.predict()
DCN.training
DeepFM.calculate_loss()
DeepFM.forward()
DeepFM.predict()
DeepFM.training
DSSM.calculate_loss()
DSSM.forward()
DSSM.predict()
DSSM.training
FFM.calculate_loss()
FFM.forward()
FFM.get_ffm_input()
FFM.predict()
FFM.training
FieldAwareFactorizationMachine.forward()
FieldAwareFactorizationMachine.training
FM.calculate_loss()
FM.forward()
FM.predict()
FM.training
FNN.calculate_loss()
FNN.forward()
FNN.predict()
FNN.training
FwFM.calculate_loss()
FwFM.forward()
FwFM.fwfm_layer()
FwFM.predict()
FwFM.training
LR.calculate_loss()
LR.forward()
LR.predict()
LR.training
NFM.calculate_loss()
NFM.forward()
NFM.predict()
NFM.training
InnerProductLayer.forward()
InnerProductLayer.training
OuterProductLayer.forward()
OuterProductLayer.training
PNN.calculate_loss()
PNN.forward()
PNN.predict()
PNN.reg_loss()
PNN.training
WideDeep.calculate_loss()
WideDeep.forward()
WideDeep.predict()
WideDeep.training
xDeepFM.calculate_loss()
xDeepFM.calculate_reg_loss()
xDeepFM.compressed_interaction_network()
xDeepFM.forward()
xDeepFM.predict()
xDeepFM.reg_loss()
xDeepFM.training