You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
from logging import getLogger
from recbole.config import Config
from recbole.data import create_dataset, data_preparation
from recbole.model.sequential_recommender import DIEN
from recbole.trainer import Trainer
from recbole.utils import init_seed, init_logger
from logging import getLogger
from recbole.config import Config
from recbole.data import create_dataset, data_preparation
from recbole.model.sequential_recommender import DIEN
from recbole.trainer import Trainer
from recbole.utils import init_seed, init_logger
if name == 'main':
config["eval_args"]
#{'split': {'LS': 'valid_and_test'},
'order': 'TO',
#'group_by': 'user',
#'mode': {'valid': 'full', 'test': 'full'}}
dataset.config["eval_args"]
#{'split': {'RS': [0.8, 0.1, 0.1]},
'order': 'RO',
'group_by': 'user',
#'mode': {'valid': 'full', 'test': 'full'}}
#原config的结果与dataset中config的结果不同,这导致在后续的train_data, valid_data, test_data = data_preparation(config, dataset)语#句中出现bug
#The ordering args for sequential recommendation has to be 'TO'
#同理在使用api快速启动模块时也出现了这个报错
#下面是我的配置文件内容
gpu_id: "0,1"
seed: 1234
encoding: 'utf-8'
data_path: /kaggle/input
show_progress: False
save_dataset: True
#数据
field_separator: "\t"
seq_separator: " "
USER_ID_FIELD: user_id
ITEM_ID_FIELD: item_id
RATING_FIELD: rating
TIME_FIELD: timestamp
seq_len: None
LABEL_FIELD: label
MAX_ITEM_LIST_LENGTH: 50
load_col:
inter: [user_id, item_id,rating, timestamp]
item: [item_id,release_year,class]
user: [user_id,age,gender,occupation,zip_code]
#训练
epochs: 100
train_batch_size: 1024
learner: 'sgd'
learning_rate: 0.01
train_neg_sample_args:
distribution: popularity
sample_num: 10
eval_step: 1
weight_decay: 1
#评估
eval_args:
split: {'LS': 'valid_and_test'}
mode: full
order: TO
repeatable: True
metrics: ['Recall', 'MRR', 'NDCG', 'Hit', 'Precision']
topk: 10
valid_metric: 'MRR@10'
eval_batch_size: 1024
#模型
alpha: 0.8
dropout_prob: 0.3
The text was updated successfully, but these errors were encountered: