wip numeric encoder as target

mindsdb · Mar 15, 2024 · 3cca968 · 3cca968
1 parent 5d06b98
commit 3cca968
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lightwood/encoder/numeric/numeric.py b/lightwood/encoder/numeric/numeric.py
@@ -40,10 +40,9 @@ def __init__(self, data_type: dtype = None,
         self.output_size = 4 if not self.is_target else 3
 
         # Weight-balance info if encoder represents target
-        self.target_weights = target_weights
-        print("Target weights: ", self.target_weights)
+        self.target_weights = None
         self.index_weights = None
-        if self.is_target:
+        if self.is_target and self.target_weights is not None:
             self.target_weights = dc(target_weights)
             self.index_weights = torch.tensor(list(self.target_weights.values()))