Start with the original variables

LLNL · Dec 2, 2024 · 69b9943 · 69b9943
1 parent 331a240
commit 69b9943
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/systems2atoms/hyperplanetree/hyperplanetree/omlt.py b/systems2atoms/hyperplanetree/hyperplanetree/omlt.py
@@ -35,7 +35,7 @@ def __init__(
             else:
                 input_bounds[i] = (-max_bound, max_bound)
 
-        used_cols = []
+        used_cols = list(range(len(input_bounds_matrix)))
         summary = copy.deepcopy(lt_regressor.summary())
         for node in summary.values():
             if isinstance(node['models'], TorchLinearRegression):