From 2e5ba57648400af2d16c6586b13eda472873cda6 Mon Sep 17 00:00:00 2001 From: Dakota Havel Date: Wed, 18 May 2022 12:50:58 -0400 Subject: [PATCH] add model metadata for dct default scores and remove unused variable in startup --- gamechangerml/api/fastapi/routers/startup.py | 1 - gamechangerml/src/search/doc_compare/model.py | 3 +++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/gamechangerml/api/fastapi/routers/startup.py b/gamechangerml/api/fastapi/routers/startup.py index 0ac2bab8..28c0fdf0 100644 --- a/gamechangerml/api/fastapi/routers/startup.py +++ b/gamechangerml/api/fastapi/routers/startup.py @@ -71,7 +71,6 @@ async def check_health(): logger.info(f"-- Sentence index name: {SENT_INDEX_PATH.value}") logger.info(f"-- QE model name: {QEXP_MODEL_NAME.value}") logger.info(f"-- QE JBOOK model name: {QEXP_JBOOK_MODEL_NAME.value}") - logger.info(f"-- QA model name: {new_qa_model_name}") logger.info(f"-- Topics model name: {TOPICS_MODEL.value}") logger.info( f"-- Doc Compare Similarity model name: {latest_doc_compare_sim.value}") diff --git a/gamechangerml/src/search/doc_compare/model.py b/gamechangerml/src/search/doc_compare/model.py index 0c48b8db..76efdd4f 100644 --- a/gamechangerml/src/search/doc_compare/model.py +++ b/gamechangerml/src/search/doc_compare/model.py @@ -223,6 +223,9 @@ def re_rank(self, query, top_docs): DEFAULT_SCORES = [ [0.8, "High"], [0.5, "Medium"], [0.4, "Low"], [0.0, "Very Low"] ] +# Metadata for the model these scores we're derived from +#{"user": null, "date_started": "2022-04-29 16:06:06", "date_finished": "2022-04-29 19:52:52", "doc_id_count": 1495122, "corpus_name": "/opt/app-root/src/gamechangerml/corpus", "encoder_model": "multi-qa-MiniLM-L6-cos-v1"} + DEFAULT_CUTOFF = 0.25