recbole.modelΒΆ
- recbole.model.context_aware_recommender
- AFM
AFM- AutoInt
AutoInt- DCN
DCN- DCN V2
DCNV2- DeepFM
DeepFM- DSSM
DSSM- EulerNet
EulerInteractionLayerEulerNet- FFM
FFMFieldAwareFactorizationMachine- FiGNN
FiGNNGraphLayer- FM
FM- FNN
FNN- FwFM
FwFM- KD_DAGFM
CINCINCompCrossNetDAGFMKD_DAGFM- LR
LR- NFM
NFM- PNN
InnerProductLayerOuterProductLayerPNN- WideDeep
WideDeep- xDeepFM
xDeepFM
- recbole.model.exlib_recommender
- recbole.model.general_recommender
AsymKNNComputeSimilarity- ADMMSLIM
ADMMSLIMadd_noise()soft_threshold()zero_mean_columns()- BPR
BPR- CDAE
CDAE- ConvNCF
ConvNCFConvNCFBPRLoss- DGCF
DGCFsample_cor_samples()- DMF
DMF- DiffRec
DNNDiffRecDiffRec.SNR()DiffRec.build_histroy_items()DiffRec.calculate_for_diffusion()DiffRec.calculate_loss()DiffRec.full_sort_predict()DiffRec.get_betas()DiffRec.input_typeDiffRec.p_mean_variance()DiffRec.p_sample()DiffRec.predict()DiffRec.q_posterior_mean_variance()DiffRec.q_sample()DiffRec.reweight_loss()DiffRec.sample_timesteps()DiffRec.trainingDiffRec.update_Lt_history()
ModelMeanTypebetas_for_alpha_bar()betas_from_linear_variance()mean_flat()normal_kl()timestep_embedding()- EASE
EASE- ENMF
ENMF- FISM
FISM- GCMC
BiDecoderGCMCGcEncoderorthogonal()- ItemKNN
ComputeSimilarityItemKNN- DiffRec
AutoEncoderLDiffReccompute_loss()- LightGCN
LightGCN- LINE
LINENegSamplingLoss- MacridVAE
MacridVAE- MultiDAE
MultiDAE- MultiVAE
MultiVAE- NAIS
NAIS- NCE-PLRec
NCEPLRec- NCL
NCL- NeuMF
NeuMF- NGCF
NGCF- NNCF
NNCF- Pop
Pop- RaCT
RaCTRaCT.calculate_ac_loss()RaCT.calculate_actor_loss()RaCT.calculate_critic_loss()RaCT.calculate_loss()RaCT.calculate_ndcg()RaCT.construct_critic_input()RaCT.construct_critic_layers()RaCT.critic_forward()RaCT.forward()RaCT.full_sort_predict()RaCT.input_typeRaCT.mlp_layers()RaCT.predict()RaCT.reparameterize()RaCT.training
- Random
Random- RecVAE
CompositePriorEncoderRecVAElog_norm_pdf()swish()- SGL
SGL- SimpleX
SimpleX- SLIMElastic
SLIMElastic- SpectralCF
SpectralCF
- recbole.model.knowledge_aware_recommender
- CFKG
CFKGInnerProductLoss- CKE
CKE- KGAT
AggregatorKGAT- KGCN
KGCN- KGIN
AggregatorGraphConvKGIN- KGNNLS
KGNNLSKGNNLS.aggregate()KGNNLS.calculate_loss()KGNNLS.calculate_ls_loss()KGNNLS.construct_adj()KGNNLS.forward()KGNNLS.full_sort_predict()KGNNLS.get_interaction_table()KGNNLS.get_neighbors()KGNNLS.input_typeKGNNLS.label_smoothness_predict()KGNNLS.predict()KGNNLS.sample_neg_interaction()KGNNLS.training
- KTUP
KTUPalignLoss()orthogonalLoss()- MCCLK
AggregatorGraphConvMCCLK- MKR
CrossCompressUnitMKR- RippleNet
RippleNet
- recbole.model.sequential_recommender
- BERT4Rec
BERT4Rec- Caser
Caser- CORE
CORETransNet- DIN
DIN- FDSA
FDSA- FOSSIL
FOSSIL- FPMC
FPMC- GCSAN
GCSANGNN- GRU4Rec
GRU4Rec- GRU4RecF
GRU4RecF- GRU4RecKG
GRU4RecKG- HGN
HGN- HRM
HRM- KSR
KSR- NARM
NARM- NextItNet
NextItNetResidualBlock_aResidualBlock_b- NPE
NPE- RepeatNet
Explore_Recommendation_DecoderRepeatNetRepeat_Explore_MechanismRepeat_Recommendation_Decoder- S3Rec
S3Rec- SASRec
SASRec- SASRecF
SASRecF- SHAN
SHAN- SRGNN
GNNSRGNN- STAMP
STAMP- TransRec
TransRec- FEARec
FEABlockFEAEncoderFEARecFEARec.alignment()FEARec.calculate_loss()FEARec.decompose()FEARec.forward()FEARec.full_sort_predict()FEARec.get_attention_mask()FEARec.get_bi_attention_mask()FEARec.get_same_item_index()FEARec.info_nce()FEARec.mask_correlated_samples()FEARec.predict()FEARec.trainingFEARec.truncated_normal_()FEARec.uniformity()
FeedForwardHybridAttention- GRU4Rec + Softmax-CPR
GRU4RecCPRgelu()- SASRec + Softmax-CPR
SASRecCPRgelu()
- recbole.model.abstract_recommender
AbstractRecommenderAutoEncoderMixinContextRecommenderContextRecommender.concat_embed_input_fields()ContextRecommender.double_tower_embed_input_fields()ContextRecommender.embed_float_fields()ContextRecommender.embed_float_seq_fields()ContextRecommender.embed_input_fields()ContextRecommender.embed_token_fields()ContextRecommender.embed_token_seq_fields()ContextRecommender.input_typeContextRecommender.trainingContextRecommender.type
GeneralRecommenderKnowledgeRecommenderSequentialRecommender- recbole.model.init
xavier_normal_initialization()xavier_uniform_initialization()- recbole.model.layers
AttLayerBaseFactorizationMachineBiGNNLayerCNNLayersContextSeqEmbAbstractLayerContextSeqEmbAbstractLayer.embed_float_fields()ContextSeqEmbAbstractLayer.embed_float_seq_fields()ContextSeqEmbAbstractLayer.embed_input_fields()ContextSeqEmbAbstractLayer.embed_token_fields()ContextSeqEmbAbstractLayer.embed_token_seq_fields()ContextSeqEmbAbstractLayer.forward()ContextSeqEmbAbstractLayer.get_embedding()ContextSeqEmbAbstractLayer.get_fields_name_dim()ContextSeqEmbAbstractLayer.training
ContextSeqEmbLayerDiceFLEmbeddingFMEmbeddingFMFirstOrderLinearFeatureSeqEmbLayerFeedForwardItemToInterestAggregationLightMultiHeadAttentionLightTransformerEncoderLightTransformerLayerMLPLayersMultiHeadAttentionSequenceAttLayerSparseDropoutTransformerEncoderTransformerLayerVanillaAttentionactivation_layer()- recbole.model.loss
BPRLossEmbLossEmbMarginLossRegLoss