nexusbert commited on
Commit
ff602f5
·
1 Parent(s): 652302c

Increase top_n_final parameter in RecommendationRequest for improved recommendation results

Browse files
Files changed (2) hide show
  1. app/main.py +2 -2
  2. app/recommendation_pipeline.py +1 -1
app/main.py CHANGED
@@ -148,7 +148,7 @@ class RecommendationRequest(BaseModel):
148
  "state": None,
149
  "chat_history": [],
150
  "top_k_retrieval": 20,
151
- "top_n_final": 3,
152
  }
153
  ]
154
  }
@@ -158,7 +158,7 @@ class RecommendationRequest(BaseModel):
158
  state: str | None = None
159
  chat_history: list[ChatTurn] = Field(default_factory=list)
160
  top_k_retrieval: int = Field(20, ge=5, le=200)
161
- top_n_final: int = Field(3, ge=1, le=25)
162
 
163
 
164
  class RecommendationRank(BaseModel):
 
148
  "state": None,
149
  "chat_history": [],
150
  "top_k_retrieval": 20,
151
+ "top_n_final": 5,
152
  }
153
  ]
154
  }
 
158
  state: str | None = None
159
  chat_history: list[ChatTurn] = Field(default_factory=list)
160
  top_k_retrieval: int = Field(20, ge=5, le=200)
161
+ top_n_final: int = Field(5, ge=1, le=25)
162
 
163
 
164
  class RecommendationRank(BaseModel):
app/recommendation_pipeline.py CHANGED
@@ -11,7 +11,7 @@ from pathlib import Path
11
  from typing import Any
12
 
13
  import numpy as np
14
-
15
  from app._paths import submission_root
16
  from app.gemini_client import gemini_generate_text, use_gemini
17
  from app.shared_models import (
 
11
  from typing import Any
12
 
13
  import numpy as np
14
+ #modules
15
  from app._paths import submission_root
16
  from app.gemini_client import gemini_generate_text, use_gemini
17
  from app.shared_models import (