From c58ba06461f4b6ea4807d4e273b9a5c570afa486 Mon Sep 17 00:00:00 2001 From: Gaurav S Deshmukh Date: Mon, 25 Sep 2023 16:59:52 -0400 Subject: [PATCH] Fixed comments --- src/train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/train.py b/src/train.py index a105c74..150ce5c 100644 --- a/src/train.py +++ b/src/train.py @@ -372,7 +372,6 @@ def predict(self, dataset, indices, return_targets=False): contributions[i, :] = ( self.standardizer.restore_cont(conts_std).numpy().flatten() ) - # contributions[i, :] = pred_dict["contributions"].cpu().detach().numpy().flatten() predictions_dict = { "targets": targets,