Views
No views yet
1ENTITY_ID_FIELD: entity_id
2HEAD_ENTITY_ID_FIELD: head_id
3ITEM_ID_FIELD: item_id
4ITEM_LIST_LENGTH_FIELD: item_length
5LABEL_FIELD: label
6LIST_SUFFIX: _list
7MAX_ITEM_LIST_LENGTH: 50
8MODEL_INPUT_TYPE: InputType.POINTWISE
9MODEL_TYPE: ModelType.SEQUENTIAL
10NEG_PREFIX: neg_
11POSITION_FIELD: position_id
12RATING_FIELD: rating
13RELATION_ID_FIELD: relation_id
14TAIL_ENTITY_ID_FIELD: tail_id
15TIME_FIELD: timestamp
16USER_ID_FIELD: user_id
17additional_feat_suffix: null
18alias_of_entity_id: null
19alias_of_item_id: null
20alias_of_relation_id: null
21alias_of_user_id: null
22attn_dropout_prob: 0.5
23benchmark_filename: null
24checkpoint_dir: saved
25clip_grad_norm: null
26dataloaders_save_path: null
27dataset: ml-100k
28dataset_save_path: saved/ml-100k.pth
29discretization: null
30enable_amp: false
31enable_scaler: false
32entity_kg_num_interval: null
33epochs: 10
34eval_args:
35 group_by: user
36 mode:
37 test: full
38 valid: full
39 order: TO
40 split:
41 LS: valid_and_test
42eval_batch_size: 256
43eval_step: 1
44eval_type: EvaluatorType.RANKING
45field_separator: "\t"
46filter_inter_by_user_or_item: true
47gpu_id: '0'
48group_size: 16
49hidden_act: gelu
50hidden_dropout_prob: 0.5
51hidden_size: 16
52initializer_range: 0.02
53inner_size: 256
54item_inter_num_interval: '[10,inf)'
55kg_reverse_r: false
56kl_beta: 0.1
57layer_norm_eps: 1.0e-12
58learner: adamw
59learning_rate: 0.0001
60load_col:
61 inter:
62 - user_id
63 - item_id
64 - rating
65 - timestamp
66local_rank: 0
67log_wandb: false
68loss_decimal_place: 4
69loss_type: CE
70metric_decimal_place: 4
71metrics:
72- NDCG
73- Hit
74- AveragePopularity
75model: SASRec
76n_heads: 2
77n_layers: 2
78normalize_all: true
79normalize_field: null
80numerical_features: []
81preload_weight: null
82relation_kg_num_interval: null
83repeatable: true
84reproducibility: true
85require_pow: false
86rm_dup_inter: null
87save_dataloaders: false
88save_dataset: true
89seed: 2020
90seq_len: null
91seq_separator: ' '
92show_progress: false
93shuffle: true
94single_spec: true
95state: INFO
96stopping_step: 10
97teacher_repo_id: viberec/ml-100k-SASRec
98test_neg_sample_args:
99 distribution: uniform
100 sample_num: none
101threshold: null
102topk:
103- 10
104train_batch_size: 256
105train_neg_sample_args:
106 alpha: none
107 candidate_num: 0
108 distribution: none
109 dynamic: false
110 sample_num: none
111transform: null
112unload_col: null
113unused_col:
114 inter:
115 - rating
116use_gpu: true
117user_inter_num_interval: '[10,inf)'
118val_interval:
119 rating: '[3,inf)'
120valid_metric: NDCG@10
121valid_metric_bigger: true
122valid_neg_sample_args:
123 distribution: uniform
124 sample_num: none
125wandb_project: recbole
126weight_decay: 0.01
127worker: 0
128