viberec/ML-100k-SASRec

SASRec trained on ml-100k

Model Description

• Model: SASRec
• Dataset: ml-100k

Performance

• ndcg@10: 0.0731
• hit@10: 0.1523
• averagepopularity@10: 164.8729

Configuration

ENTITY_ID_FIELD: entity_id
HEAD_ENTITY_ID_FIELD: head_id
ITEM_ID_FIELD: item_id
ITEM_LIST_LENGTH_FIELD: item_length
LABEL_FIELD: label
LIST_SUFFIX: _list
MAX_ITEM_LIST_LENGTH: 50
MODEL_INPUT_TYPE: InputType.POINTWISE
MODEL_TYPE: ModelType.SEQUENTIAL
NEG_PREFIX: neg_
POSITION_FIELD: position_id
RATING_FIELD: rating
RELATION_ID_FIELD: relation_id
TAIL_ENTITY_ID_FIELD: tail_id
TIME_FIELD: timestamp
USER_ID_FIELD: user_id
additional_feat_suffix: null
alias_of_entity_id: null
alias_of_item_id: null
alias_of_relation_id: null
alias_of_user_id: null
attn_dropout_prob: 0.5
benchmark_filename: null
checkpoint_dir: saved
clip_grad_norm: null
dataloaders_save_path: null
dataset: ml-100k
dataset_save_path: saved/ml-100k.pth
discretization: null
enable_amp: false
enable_scaler: false
entity_kg_num_interval: null
epochs: 500
eval_args:
  group_by: user
  mode:
    test: full
    valid: full
  order: TO
  split:
    LS: valid_and_test
eval_batch_size: 256
eval_step: 1
eval_type: EvaluatorType.RANKING
field_separator: "\t"
filter_inter_by_user_or_item: true
gpu_id: '0'
hidden_act: gelu
hidden_dropout_prob: 0.5
hidden_size: 64
initializer_range: 0.02
inner_size: 256
item_inter_num_interval: '[10,inf)'
kg_reverse_r: false
layer_norm_eps: 1.0e-12
learner: adam
learning_rate: 0.003
load_col:
  inter:
  - user_id
  - item_id
  - rating
  - timestamp
local_rank: 0
log_wandb: false
loss_decimal_place: 4
loss_type: CE
metric_decimal_place: 4
metrics:
- NDCG
- Hit
- AveragePopularity
model: SASRec
n_heads: 2
n_layers: 2
normalize_all: true
normalize_field: null
numerical_features: []
preload_weight: null
relation_kg_num_interval: null
repeatable: true
reproducibility: true
require_pow: false
rm_dup_inter: null
save_dataloaders: false
save_dataset: true
seed: 2020
seq_len: null
seq_separator: ' '
show_progress: false
shuffle: true
single_spec: true
state: INFO
stopping_step: 10
test_neg_sample_args:
  distribution: uniform
  sample_num: none
threshold: null
topk:
- 10
train_batch_size: 256
train_neg_sample_args:
  alpha: none
  candidate_num: 0
  distribution: none
  dynamic: false
  sample_num: none
transform: null
unload_col: null
unused_col:
  inter:
  - rating
use_gpu: true
user_inter_num_interval: '[10,inf)'
val_interval:
  rating: '[3,inf)'
valid_metric: NDCG@10
valid_metric_bigger: true
valid_neg_sample_args:
  distribution: uniform
  sample_num: none
wandb_project: recbole
weight_decay: 0.0
worker: 0