jibala-1022
commited on
Commit
โข
1fdeae3
1
Parent(s):
84a7b61
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,38 @@
|
|
1 |
-
---
|
2 |
-
license:
|
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 | Model Size | Model Parameters | Recall@10 on Movies |
|
26 |
+
|:-------------------------------|:--------:| :--------:|
|
27 |
+
| [jibala-1022/easyrec-small](https://huggingface.co/jibala-1022/easyrec-roberta-small) | 243M | 121,364,313 | 0.0286 |
|
28 |
+
| [jibala-1022/easyrec-base](https://huggingface.co/jibala-1022/easyrec-roberta-base) | 328M | 163,891,545 | 0.0023 |
|
29 |
+
|
30 |
+
## ๐ Citation
|
31 |
+
```bibtex
|
32 |
+
@article{ren2024easyrec,
|
33 |
+
title={EasyRec: Simple yet Effective Language Models for Recommendation},
|
34 |
+
author={Ren, Xubin and Huang, Chao},
|
35 |
+
journal={arXiv preprint arXiv:2408.08821},
|
36 |
+
year={2024}
|
37 |
+
}
|
38 |
+
```
|