sasrec

Classes

SASRecDataPreparator(session_max_len, ...[, ...])	Data preparator for SASRecModel.
SASRecModel(n_blocks, n_heads, n_factors, ...)	SASRec model: transformer-based sequential model with unidirectional attention mechanism and "Shifted Sequence" training objective.
SASRecModelConfig(*, cls, verbose, ...)	SASRecModel config.
SASRecTransformerLayer(n_factors, n_heads, ...)	Exactly SASRec author's transformer block architecture but with pytorch Multi-Head Attention realisation.
SASRecTransformerLayers(n_blocks, n_factors, ...)	SASRec transformer blocks.