From f227b1875870b4b466a21c09f7fa0cacf4325b2a Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Wed, 15 Apr 2020 18:04:12 -0500 Subject: [PATCH] Primal: add progress bar --- src/python/miplearn/components/primal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/python/miplearn/components/primal.py b/src/python/miplearn/components/primal.py index 26bb35b..3ad7046 100644 --- a/src/python/miplearn/components/primal.py +++ b/src/python/miplearn/components/primal.py @@ -46,7 +46,7 @@ class PrimalSolutionComponent(Component): features = VariableFeaturesExtractor().extract(training_instances) solutions = SolutionExtractor().extract(training_instances) - for category in features.keys(): + for category in tqdm(features.keys(), desc="Fit (primal)"): x_train = features[category] for label in [0, 1]: y_train = solutions[category][:, label].astype(int)