Fixed comments

GreeleyGroup · Sep 25, 2023 · c58ba06 · c58ba06
1 parent b8c319b
commit c58ba06
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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,