Extract all features ahead of time

2025-12-06 01:18:52 -06:00 · 2021-03-31 07:42:01 -05:00
parent b3c24814b0
commit 12fca1f22b
8 changed files with 106 additions and 39 deletions
--- a/miplearn/components/primal.py
+++ b/miplearn/components/primal.py
@@ -236,7 +236,7 @@ class PrimalSolutionComponent(Component):
            PrimalSolutionComponent._extract(
                instance,
                sample,
-                instance.model_features["Variables"],
+                instance.features["Variables"],
                extract_y=False,
            ),
        )