求问：使用auc和logloss指标效果很差

RecBole-CDR

RecBole-CDR is a library built upon RecBole for reproducing and developing cross-domain recommendation algorithms.

Highlights

Automatic and compatible data processing for cross-domain recommendation: Our library designs a unified data structure for cross-domain recommendation, which inherits all the data pre-processing strategies in RecBole. The overlapped data in different domains can be matched automatically.
Flexible and customized model training strategies: Our library provides four basic training modes for cross-domain recommendation, which can be combined arbitrarily by users. It is also easy to customize training strategy in original way.
Extensive cross-domain recommendation algorithms: Based on unified data structure and flexible training strategies, several cross-domain recommendation algorithms are implemented and compared with others fairly.

Requirements

recbole==1.0.1
torch>=1.7.0
python>=3.7.0

Quick-Start

With the source code, you can use the provided script for initial usage of our library:

python run_recbole_cdr.py

This script will run the CMF model with ml-1m as source domain dataset and ml-100k as target domain dataset.

If you want to change the models, just run the script by setting additional command parameters:

python run_recbole_cdr.py --model=[model]

Implemented Models

We list currently supported Cross-Domain Recommendation models:

CMF from Singh et al.: Relational Learning via Collective Matrix Factorization (SIGKDD 2008).
DTCDR from Zhu et al.: DTCDR: A Framework for Dual-Target Cross-Domain Recommendation (CIKM 2019).
CoNet from Hu et al.: CoNet: Collaborative Cross Networks for Cross-Domain Recommendation (CIKM 2018).
BiTGCF from Liu et al.: Cross Domain Recommendation via Bi-directional Transfer Graph Collaborative Filtering Networks (CIKM 2020).
CLFM from Gao et al.: Cross-Domain Recommendation via Cluster-Level Latent Factor Model (PKDD 2013).
DeepAPF from Yan et al.: DeepAPF: Deep Attentive Probabilistic Factorization for Multi-site Video Recommendation (IJCAI 2019).
NATR from Gao et al.: Cross-domain Recommendation Without Sharing User-relevant Data (WWW 2019).
EMCDR from Man et al.: Cross-Domain Recommendation: An Embedding and Mapping Approach (IJCAI 2017).
SSCDR from Kang et al.: Semi-Supervised Learning for Cross-Domain Recommendation to Cold-Start Users (CIKM 2019).
DCDCSR from Zhu et al.: A Deep Framework for Cross-Domain and Cross-System Recommendations (IJCAI 2018).

Result

Dataset

We collected and organized three pairs of datasets with one source domain and one target domain which are commonly used in cross-domain recommendation. Here we provide these datasets for reference:

Amazon datasets;
Book-Crossing datasets;
Douban datasets;

Hyper-parameters

We carefully tune the hyper-parameters of the implemented models on these datasets and we provide these hyper-parameters here for reference:

Cross-domain-recommendation on Amazon datasets;
Cross-domain-recommendation on Book-Crossing datasets;
Cross-domain-recommendation on Douban datasets;

Contributing

Please let us know if you encounter a bug or have any suggestions by filing an issue.

We welcome all contributions from bug fixes to new features and extensions.

We expect all contributions discussed in the issue tracker and going through PRs.

The Team

RecBole-CDR is developed and maintained by members from RUCAIBox, the main developers are Zihan Lin (@linzihan-backforward), Gaowei Zhang (@Wicknight) and Shanlei Mu (@ShanleiMu).

Acknowledgement

The implementation is based on the open-source recommendation library RecBole.

Please cite the following paper as the reference if you use our code or processed datasets.

@inproceedings{zhao2021recbole,
  title={Recbole: Towards a unified, comprehensive and efficient framework for recommendation algorithms},
  author={Wayne Xin Zhao and Shanlei Mu and Yupeng Hou and Zihan Lin and Kaiyuan Li and Yushuo Chen and Yujie Lu and Hui Wang and Changxin Tian and Xingyu Pan and Yingqian Min and Zhichao Feng and Xinyan Fan and Xu Chen and Pengfei Wang and Wendi Ji and Yaliang Li and Xiaoling Wang and Ji-Rong Wen},
  booktitle={{CIKM}},
  year={2021}
}

	with torch.no_grad():
	self.source_user_embedding.weight[self.overlapped_num_users: self.target_num_users].fill_(0)
	self.source_item_embedding.weight[self.overlapped_num_items: self.target_num_items].fill_(0)

	self.target_user_embedding.weight[self.target_num_users:].fill_(0)
	self.target_item_embedding.weight[self.target_num_items:].fill_(0)

	if not source_split_flag:
	source_domain_train_dataset = self.source_domain_dataset
	source_domain_train_dataset._change_feat_format()

	return [source_domain_train_dataset, None, target_domain_train_dataset,
	target_domain_valid_dataset, target_domain_test_dataset]
	else:
	source_domain_train_dataset, source_domain_valid_dataset = self.source_domain_dataset.split_train_valid()
	return [source_domain_train_dataset, source_domain_valid_dataset, target_domain_train_dataset,
	target_domain_valid_dataset, target_domain_test_dataset]

rucaibox / recbole-cdr Goto Github PK

recbole-cdr's Introduction

RecBole-CDR

Highlights

Requirements

Quick-Start

Implemented Models

Result

Dataset

Hyper-parameters

Contributing

The Team

Acknowledgement

recbole-cdr's People

Contributors

Stargazers

Watchers

Forkers

recbole-cdr's Issues

Recommend Projects

Recommend Topics

Recommend Org