From f687b6edad4a8fed66f9ae825b438978d3a8e627 Mon Sep 17 00:00:00 2001 From: Joppe Geluykens Date: Mon, 30 Jan 2023 19:29:50 +0100 Subject: [PATCH] Explain: improve docstring about IntegratedGradient baseline for number features (#3018) --- ludwig/explain/captum.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ludwig/explain/captum.py b/ludwig/explain/captum.py index 3f27ea2d645..e19e7d7689a 100644 --- a/ludwig/explain/captum.py +++ b/ludwig/explain/captum.py @@ -284,8 +284,7 @@ def get_baseline(model: LudwigModel, sample_encoded: List[Variable]) -> List[tor elif feature.type() == IMAGE: baseline = torch.zeros_like(sample_input[0], device=DEVICE) else: - # For a robust baseline, we take the mean of all embeddings in the sample from the training data. - # TODO(joppe): now that we don't have embeddings, we should re-evaluate this. + # For a robust baseline, we take the mean of all samples from the training data. baseline = torch.mean(sample_input.float(), dim=0) baselines.append(baseline.unsqueeze(0))