Research Article

An Improved Sequential Recommendation Algorithm based on Short-Sequence Enhancement and Temporal Self-Attention Mechanism

Table 2

The parameters of the proposed model and the experimental environment.

ModelLearning rate0.001
Momentum0.9
Dropout rate0.2
Batch size128
Maximum iterations200
Validation interval20
Regularization0.00005
Short-sequence threshold20
Maximum sequence length for ML-1M70
Maximum sequence length for AM-BE30
Latent dimension for ML-1M50
Latent dimension for AM-BE20
Pseudo-historical item for ML-1M5
Pseudo-historical item for AM-BE15

EnvironmentProgramming softwarePython3.6
Deep learning frameworkPytorch
Computer systemWindows 10
CpuE5-2620 v4
RAM32.0 GB
GpuGeForce RTX 2080