Merge pull request #129 from dod-advana/bugfix/add-dct-score-metadata

add model metadata for dct default scores and remove unused variable …
dod-advana · May 31, 2022 · b4fe0bd · b4fe0bd
2 parents 9948adb + 2e5ba57
commit b4fe0bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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