jibala-1022
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,39 @@
|
|
1 |
-
---
|
2 |
-
license: apache-2.0
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
language: en
|
4 |
+
tags:
|
5 |
+
- recommendation
|
6 |
+
- collaborative filtering
|
7 |
+
metrics: recall@10
|
8 |
+
base_model: FacebookAI/roberta-base
|
9 |
+
pipeline_tag: sentence-similarity
|
10 |
+
---
|
11 |
+
|
12 |
+
# EasyRec-Base
|
13 |
+
|
14 |
+
## Overview
|
15 |
+
|
16 |
+
- **Description**: EasyRec is a series of language models designed for recommendations, trained to match the textual profiles of users and items with collaborative signals.
|
17 |
+
- **Usage**: You can use EasyRec to encode user and item text embeddings based on the textual profiles that reflect their preferences for various recommendation scenarios.
|
18 |
+
- **Evaluation**: We evaluate the performance of EasyRec in: (i) Text-based zero-shot recommendation and (ii) Text-enhanced collaborative filtering.
|
19 |
+
- **Finetuned from model:** EasyRec is finetuned from [RoBERTa](https://huggingface.co/FacebookAI/roberta-large) within English.
|
20 |
+
|
21 |
+
For details please refer [💻[GitHub Code](https://github.com/jibala-1022/EasyRec)] and [📖[Paper](https://arxiv.org/abs/2408.08821)].
|
22 |
+
|
23 |
+
### Model List
|
24 |
+
We release a series of EasyRec checkpoints with varying sizes. You can easily load these models from Hugging Face by replacing the model name.
|
25 |
+
| Model | Size | Parameters | Recall@10 on Movies |
|
26 |
+
|:-----:|:----:|:----------:|:-------------------:|
|
27 |
+
| [jibala-1022/easyrec-small](https://huggingface.co/jibala-1022/easyrec-small) | 243 MB | 121,364,313 | 0.0086 |
|
28 |
+
| [jibala-1022/easyrec-base](https://huggingface.co/jibala-1022/easyrec-base) | 328 MB | 163,891,545 | 0.0166 |
|
29 |
+
| [jibala-1022/easyrec-large](https://huggingface.co/jibala-1022/easyrec-large) | 816 MB | 407,933,017 | 0.0166 |
|
30 |
+
|
31 |
+
## 🌟 Citation
|
32 |
+
```bibtex
|
33 |
+
@article{ren2024easyrec,
|
34 |
+
title={EasyRec: Simple yet Effective Language Models for Recommendation},
|
35 |
+
author={Ren, Xubin and Huang, Chao},
|
36 |
+
journal={arXiv preprint arXiv:2408.08821},
|
37 |
+
year={2024}
|
38 |
+
}
|
39 |
+
```
|