Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,154 @@
+---
+tags:
+- viberec
+- recommender-system
+- sasrec
+datasets:
+- ml-100k
+metrics:
+averagepopularity@10: 199.4036
+hit@10: 0.1374
+ndcg@10: 0.0631
+
+---
+
+# SASRec trained on ml-100k
+
+
+## Model Description
+- **Model**: SASRec
+- **Dataset**: ml-100k
+
+## Performance
+- **ndcg@10**: 0.0631
+- **hit@10**: 0.1374
+- **averagepopularity@10**: 199.4036
+
+## Configuration
+```yaml
+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: 16
+initializer_range: 0.02
+inner_size: 256
+item_inter_num_interval: '[10,inf)'
+kg_reverse_r: false
+layer_norm_eps: 1.0e-12
+learner: adamw
+learning_rate: 0.001
+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.01
+worker: 0
+
+```