recbole.modelΒΆ
- recbole.model.context_aware_recommender
- AFM
AFM
- AutoInt
AutoInt
- DCN
DCN
- DCN V2
DCNV2
- DeepFM
DeepFM
- DSSM
DSSM
- EulerNet
EulerInteractionLayer
EulerNet
- FFM
FFM
FieldAwareFactorizationMachine
- FiGNN
FiGNN
GraphLayer
- FM
FM
- FNN
FNN
- FwFM
FwFM
- KD_DAGFM
CIN
CINComp
CrossNet
DAGFM
KD_DAGFM
- LR
LR
- NFM
NFM
- PNN
InnerProductLayer
OuterProductLayer
PNN
- WideDeep
WideDeep
- xDeepFM
xDeepFM
- recbole.model.exlib_recommender
- recbole.model.general_recommender
AsymKNN
ComputeSimilarity
- ADMMSLIM
ADMMSLIM
add_noise()
soft_threshold()
zero_mean_columns()
- BPR
BPR
- CDAE
CDAE
- ConvNCF
ConvNCF
ConvNCFBPRLoss
- DGCF
DGCF
sample_cor_samples()
- DMF
DMF
- DiffRec
DNN
DiffRec
DiffRec.SNR()
DiffRec.build_histroy_items()
DiffRec.calculate_for_diffusion()
DiffRec.calculate_loss()
DiffRec.full_sort_predict()
DiffRec.get_betas()
DiffRec.input_type
DiffRec.p_mean_variance()
DiffRec.p_sample()
DiffRec.predict()
DiffRec.q_posterior_mean_variance()
DiffRec.q_sample()
DiffRec.reweight_loss()
DiffRec.sample_timesteps()
DiffRec.training
DiffRec.update_Lt_history()
ModelMeanType
betas_for_alpha_bar()
betas_from_linear_variance()
mean_flat()
normal_kl()
timestep_embedding()
- EASE
EASE
- ENMF
ENMF
- FISM
FISM
- GCMC
BiDecoder
GCMC
GcEncoder
orthogonal()
- ItemKNN
ComputeSimilarity
ItemKNN
- DiffRec
AutoEncoder
LDiffRec
compute_loss()
- LightGCN
LightGCN
- LINE
LINE
NegSamplingLoss
- MacridVAE
MacridVAE
- MultiDAE
MultiDAE
- MultiVAE
MultiVAE
- NAIS
NAIS
- NCE-PLRec
NCEPLRec
- NCL
NCL
- NeuMF
NeuMF
- NGCF
NGCF
- NNCF
NNCF
- Pop
Pop
- RaCT
RaCT
RaCT.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_type
RaCT.mlp_layers()
RaCT.predict()
RaCT.reparameterize()
RaCT.training
- Random
Random
- RecVAE
CompositePrior
Encoder
RecVAE
log_norm_pdf()
swish()
- SGL
SGL
- SimpleX
SimpleX
- SLIMElastic
SLIMElastic
- SpectralCF
SpectralCF
- recbole.model.knowledge_aware_recommender
- CFKG
CFKG
InnerProductLoss
- CKE
CKE
- KGAT
Aggregator
KGAT
- KGCN
KGCN
- KGIN
Aggregator
GraphConv
KGIN
- KGNNLS
KGNNLS
KGNNLS.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_type
KGNNLS.label_smoothness_predict()
KGNNLS.predict()
KGNNLS.sample_neg_interaction()
KGNNLS.training
- KTUP
KTUP
alignLoss()
orthogonalLoss()
- MCCLK
Aggregator
GraphConv
MCCLK
- MKR
CrossCompressUnit
MKR
- RippleNet
RippleNet
- recbole.model.sequential_recommender
- BERT4Rec
BERT4Rec
- Caser
Caser
- CORE
CORE
TransNet
- DIN
DIN
- FDSA
FDSA
- FOSSIL
FOSSIL
- FPMC
FPMC
- GCSAN
GCSAN
GNN
- GRU4Rec
GRU4Rec
- GRU4RecF
GRU4RecF
- GRU4RecKG
GRU4RecKG
- HGN
HGN
- HRM
HRM
- KSR
KSR
- NARM
NARM
- NextItNet
NextItNet
ResidualBlock_a
ResidualBlock_b
- NPE
NPE
- RepeatNet
Explore_Recommendation_Decoder
RepeatNet
Repeat_Explore_Mechanism
Repeat_Recommendation_Decoder
- S3Rec
S3Rec
- SASRec
SASRec
- SASRecF
SASRecF
- SHAN
SHAN
- SRGNN
GNN
SRGNN
- STAMP
STAMP
- TransRec
TransRec
- FEARec
FEABlock
FEAEncoder
FEARec
FEARec.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.training
FEARec.truncated_normal_()
FEARec.uniformity()
FeedForward
HybridAttention
- GRU4Rec + Softmax-CPR
GRU4RecCPR
gelu()
- SASRec + Softmax-CPR
SASRecCPR
gelu()
- recbole.model.abstract_recommender
AbstractRecommender
AutoEncoderMixin
ContextRecommender
ContextRecommender.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_type
ContextRecommender.training
ContextRecommender.type
GeneralRecommender
KnowledgeRecommender
SequentialRecommender
- recbole.model.init
xavier_normal_initialization()
xavier_uniform_initialization()
- recbole.model.layers
AttLayer
BaseFactorizationMachine
BiGNNLayer
CNNLayers
ContextSeqEmbAbstractLayer
ContextSeqEmbAbstractLayer.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
ContextSeqEmbLayer
Dice
FLEmbedding
FMEmbedding
FMFirstOrderLinear
FeatureSeqEmbLayer
FeedForward
ItemToInterestAggregation
LightMultiHeadAttention
LightTransformerEncoder
LightTransformerLayer
MLPLayers
MultiHeadAttention
SequenceAttLayer
SparseDropout
TransformerEncoder
TransformerLayer
VanillaAttention
activation_layer()
- recbole.model.loss
BPRLoss
EmbLoss
EmbMarginLoss
RegLoss