ChernovAndrei commited on
Commit
c44caaa
·
1 Parent(s): 72cb97a

updated ReadME

Browse files
Files changed (1) hide show
  1. README.md +38 -12
README.md CHANGED
@@ -1,21 +1,47 @@
1
  # Movie Recommender System
2
- **Tag:** `agent-demo-track`
 
 
3
  A hybrid movie recommender system that combines collaborative filtering, language model embeddings, and graph convolutional networks to provide personalized movie recommendations.
4
 
5
  ## Features
6
 
7
- - **Dual Embedding Types:**
8
- - Pure Language Model (LLM) embeddings from Mistral AI
9
- - Graph-enhanced embeddings (LLM + GCL) that combine language understanding with user interaction patterns
10
- - **Hybrid Input:**
11
- - Select up to 5 movies you've enjoyed
12
- - Describe what kind of movie you're looking for in natural language
13
- - Adjust the weight (α) between your movie selections and text description
14
- - **Rich Results:**
15
- - Get up to 20 personalized recommendations
16
- - View similarity scores for each recommendation
17
- - Search through a database of over 100,000 movies
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
 
 
 
 
 
19
  ## Requirements
20
 
21
  1. Python 3.8+
 
1
  # Movie Recommender System
2
+
3
+ # Tag: **agent-demo-track**
4
+
5
  A hybrid movie recommender system that combines collaborative filtering, language model embeddings, and graph convolutional networks to provide personalized movie recommendations.
6
 
7
  ## Features
8
 
9
+ ### Dual Embedding Types
10
+
11
+ - **Pure Language Model (LLM) Embeddings**
12
+ Generated for each movie title using Mistral AI.
13
+
14
+ - **Graph-Enhanced Embeddings (LLM + GCL)**
15
+ Combines language understanding with user interaction patterns to enrich the embeddings.
16
+
17
+ ---
18
+
19
+ ### Hybrid Input
20
+
21
+ - **Movie Selection**
22
+ Select movies you've previously enjoyed.
23
+
24
+ - **Natural Language Query**
25
+ Describe the kind of movie you're looking for in natural language.
26
+
27
+ - **Weight Adjustment (α)**
28
+ Adjust the balance between your movie selections and your text description to personalize the recommendations.
29
+
30
+ ---
31
+
32
+ ### Algorithm
33
+
34
+ - **Embedding Aggregation**
35
+ Convert the user preference into an embedding and aggregate it with embeddings of previously watched movies to create a query embedding.
36
+
37
+ - **Retrieval Phase**
38
+ Retrieve the top 100 candidate movies based on cosine similarity between the query embedding and movie embeddings.
39
 
40
+ - **Ranking Phase**
41
+ Use an AI agent to rank the top 100 candidates and select the final top 10 recommendations, considering:
42
+ - User preferences
43
+ - Viewing history
44
+ - Weight parameter (α)
45
  ## Requirements
46
 
47
  1. Python 3.8+