RUCAIBox / RecBole-CDR

This is a library built upon RecBole for cross-domain recommendation algorithms
MIT License
82 stars 12 forks source link

Running CDR algorithms on the datasets provided #61

Open ajaykv1 opened 11 months ago

ajaykv1 commented 11 months ago

I would like to run all the 10 CDR algorithms on the Amazon and Douban datasets.

I see that you provided the hyperparameters for each model, and I'd like to use them to get results.

Is it as simple as running:

python run_recbole_cdr.py --model=[model] --dataset=Amazon

If not, How can specify the dataset and run the models with the same hyperparamter configurations that you mentioned?

Wicknight commented 9 months ago

Hi @ajaykv1 Since the training process involves multiple datasets (source datasets, target datasets), we do not support specifying with just one parameter of the dataset. In contrast, we set the 'config_files' parameter to specify the configuration file, and one can specify the details of each dataset in that file. Therefore, the standard command for training is as follows:

python run_recbole_cdr.py --model=[model] --config_files=[yaml_file]

You can refer to our example for the standard configuration of config_files.