File size: 12,951 Bytes
6fe071b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 | ---
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]
|