From 0945629c745adae221941b811acdf57f880a7e8e Mon Sep 17 00:00:00 2001 From: franz101 Date: Thu, 18 May 2023 11:56:35 -0700 Subject: [PATCH] SetFit: When logging probs, log them not as logits (#612) This is a proposed fixed that might resolve quirks around the DEP reported for setfit --- dataquality/__init__.py | 2 +- dataquality/integrations/setfit.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/dataquality/__init__.py b/dataquality/__init__.py index 40eebaf92..44afa265b 100644 --- a/dataquality/__init__.py +++ b/dataquality/__init__.py @@ -30,7 +30,7 @@ dataquality.get_insights() """ -__version__ = "v0.8.42" +__version__ = "v0.8.43" import sys from typing import Any, List, Optional diff --git a/dataquality/integrations/setfit.py b/dataquality/integrations/setfit.py index 248b64a33..450789276 100644 --- a/dataquality/integrations/setfit.py +++ b/dataquality/integrations/setfit.py @@ -246,7 +246,7 @@ def __call__(self, *args: Any, **kwds: Any) -> Any: # 🔭🌕 Galileo logging dq.log_model_outputs( ids=batch["id"], - logits=dq_store["output"], + probs=dq_store["output"], embs=dq_store["input_args"][0], split=split, epoch=0, @@ -362,7 +362,7 @@ def dq_evaluate( # 🔭🌕 Galileo logging dq.log_model_outputs( ids=batch[id_col], - logits=dq_store["output"], + probs=dq_store["output"], embs=dq_store["input_args"][0], split=split, epoch=0,