ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-07 09:58:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Primal: add progress bar

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-15 18:04:12 -05:00
parent a39159158e
commit f227b18758
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/python/miplearn/components/primal.py
View File

@@ -46,7 +46,7 @@ class PrimalSolutionComponent(Component):
features = VariableFeaturesExtractor().extract(training_instances) features = VariableFeaturesExtractor().extract(training_instances)
solutions = SolutionExtractor().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] x_train = features[category]
for label in [0, 1]: for label in [0, 1]:
y_train = solutions[category][:, label].astype(int) y_train = solutions[category][:, label].astype(int)