bert4rec

Classes

BERT4RecDataPreparator(session_max_len, ...)

Data Preparator for BERT4RecModel.

BERT4RecModel(n_blocks, n_heads, n_factors, ...)

BERT4Rec model: transformer-based sequential model with bidirectional attention mechanism and "MLM" (masked item in user sequence) training objective.

BERT4RecModelConfig(*, cls, verbose, ...)

BERT4RecModel config.