zjkarina commited on
Commit
007c064
·
verified ·
1 Parent(s): 90096b0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -19
README.md CHANGED
@@ -1,41 +1,40 @@
 
 
 
1
  # zjkarina/omniRecsysLLM_semanticIDsmodality
2
 
3
- Модель рекомендаций с семантическими ID для Amazon Fashion.
4
 
5
- ## Описание
6
 
7
- Эта модель использует VQ-VAE для создания семантических ID товаров, что позволяет более точно понимать семантические связи между товарами.
8
 
9
- ## Архитектура
10
 
11
- - **Базовая модель**: Qwen2.5-Omni-7B
12
- - **Размер словаря товаров**: 709,036
13
- - **Размерность ID эмбеддингов**: 512
14
  - **VQ-VAE codebook size**: 10,000
15
  - **VQ-VAE codebook dimension**: 256
16
- - **Датасет**: Amazon Fashion 2023 Full
17
 
18
- ## Использование
19
 
20
  ```python
21
  from any2any_trainer.models.recommendation import SemanticIDRecommendationModel
22
 
23
- # Загрузка модели
24
  model = SemanticIDRecommendationModel.from_pretrained("zjkarina/omniRecsysLLM_semanticIDsmodality")
25
 
26
- # Генерация рекомендаций с семантическими ID
27
  recommendations = model.predict_next_item(
28
- text="Пользователь купил джинсы и футболку",
29
- id_ids=[12345, 67890], # ID товаров из истории
30
  top_k=5,
31
  use_semantic_ids=True
32
  )
33
  ```
34
 
35
- ## Обучение
36
 
37
- Модель обучена на датасете Amazon Fashion 2023 с использованием семантических ID через VQ-VAE.
38
-
39
- ## Лицензия
40
-
41
- Apache-2.0 License
 
1
+ ---
2
+ license: mit
3
+ ---
4
  # zjkarina/omniRecsysLLM_semanticIDsmodality
5
 
6
+ Recommendation model with semantic IDs for Amazon Fashion.
7
 
8
+ ## Description
9
 
10
+ This model uses VQ-VAE to create semantic item IDs, enabling a more accurate understanding of semantic relationships between products.
11
 
12
+ ## Architecture
13
 
14
+ - **Base model**: Qwen2.5-Omni-7B
15
+ - **Item vocabulary size**: 709,036
16
+ - **ID embedding dimension**: 512
17
  - **VQ-VAE codebook size**: 10,000
18
  - **VQ-VAE codebook dimension**: 256
19
+ - **Dataset**: Amazon Fashion 2023 Full
20
 
21
+ ## Usage
22
 
23
  ```python
24
  from any2any_trainer.models.recommendation import SemanticIDRecommendationModel
25
 
26
+ # Load model
27
  model = SemanticIDRecommendationModel.from_pretrained("zjkarina/omniRecsysLLM_semanticIDsmodality")
28
 
29
+ # Generate recommendations with semantic IDs
30
  recommendations = model.predict_next_item(
31
+ text="The user bought jeans and a t-shirt",
32
+ id_ids=[12345, 67890], # Item IDs from purchase history
33
  top_k=5,
34
  use_semantic_ids=True
35
  )
36
  ```
37
 
38
+ ## Training
39
 
40
+ The model was trained on the Amazon Fashion 2023 dataset using semantic IDs generated via VQ-VAE.