Adjust zscore bias for descriptions (#14205)

This commit is contained in:
Josh Hawkins 2024-10-07 19:51:37 -05:00 committed by GitHub
parent 6ebad84160
commit 23ce1e930d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -73,7 +73,7 @@ class EmbeddingsContext:
def __init__(self, db: SqliteVecQueueDatabase): def __init__(self, db: SqliteVecQueueDatabase):
self.embeddings = Embeddings(db) self.embeddings = Embeddings(db)
self.thumb_stats = ZScoreNormalization() self.thumb_stats = ZScoreNormalization()
self.desc_stats = ZScoreNormalization(scale_factor=2.5, bias=0.5) self.desc_stats = ZScoreNormalization(scale_factor=3, bias=-2.5)
# load stats from disk # load stats from disk
try: try: