--- license: mit tags: - recbole - recommendation - benchmark --- # RecBole Model Checkpoints This repository contains 52 model checkpoint(s) for RecBole. ## Repository Information - **Total Models**: 52 - **Format**: PyTorch `.pth` files - **Framework**: [RecBole](https://github.com/RUCAIBox/RecBole) ## Available Models ### BPR - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_retrain_checkpoint_idx_to_match_3_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_meat.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_150_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_225_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_75_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_150_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_225_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_75_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_150_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_225_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_75_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_150_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_225_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_amazon_reviews_grocery_and_gourmet_food_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_alcohol_unlearn_epoch_75_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_goodreads_best.pth` - `model_BPR_seed_11_dataset_goodreads_retrain_checkpoint_idx_to_match_3_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1450_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_2900_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_4350_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_5799_retrain_checkpoint_idx_to_match_3.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1450_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_2900_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_4350_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_5799_retrain_checkpoint_idx_to_match_3.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1450_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_2900_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_4350_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_5799_retrain_checkpoint_idx_to_match_3.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1450_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_2900_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_4350_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_goodreads_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_5799_retrain_checkpoint_idx_to_match_3.pth` - `model_BPR_seed_11_dataset_movielens_retrain_checkpoint_idx_to_match_3_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_violence.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_15_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_30_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_fanchuan_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_45_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1462_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_15_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_30_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_45_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_487_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_60_retrain_checkpoint_idx_to_match_3.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_kookmin_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_975_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1462_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_15_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_30_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_45_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_487_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_60_retrain_checkpoint_idx_to_match_3.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_scif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_975_retrain_checkpoint_idx_to_match_1.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_1462_retrain_checkpoint_idx_to_match_2.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_15_retrain_checkpoint_idx_to_match_0.pth` - `model_BPR_seed_11_dataset_movielens_unlearning_algorithm_seif_unlearning_fraction_0.0001_unlearning_sample_selection_method_sensitive_category_health_unlearn_epoch_30_retrain_checkpoint_idx_to_match_1.pth` ## Usage with RecBole ### Method 1: Command Line ```bash python run_recbole.py \ --model YOUR_MODEL \ --dataset YOUR_DATASET \ --eval_only \ --hf_model_path "hf://deem-data/recbole-models/FILENAME.pth" ``` ### Method 2: Python API ```python from recbole.utils import load_model_from_path import torch # Load checkpoint checkpoint = load_model_from_path( model_path="hf://deem-data/recbole-models/FILENAME.pth", map_location="cpu" ) # Load into your model model.load_state_dict(checkpoint["state_dict"]) ``` ### Method 3: Download All Models ```python from recbole.utils import HuggingFaceModelLoader loader = HuggingFaceModelLoader() repo_path = loader.download_repository( repo_id="deem-data/recbole-models", allow_patterns=["*.pth"] ) print(f"All models downloaded to: {repo_path}") ``` ## Citation If you use these models, please cite: ```bibtex @inproceedings{recbole, 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 Yushuo Chen and Xingyu Pan and Kaiyuan Li and Yujie Lu and Hui Wang and Changxin Tian 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} } ``` ## License MIT License (or specify your own) ## Contact For questions about these models, please contact: [your email or GitHub]