Increase top_n_final parameter in RecommendationRequest for improved recommendation results
Browse files- app/main.py +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":
|
| 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(
|
| 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 (
|