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

Branch: add progress bar

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-24 12:42:20 -05:00
parent 80f2251877
commit e352e478d0
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -55,7 +55,7 @@ class BranchPriorityComponent(Component):
pass
def fit(self, training_instances, n_jobs=1):
for instance in training_instances:
for instance in tqdm(training_instances, desc="Fit (branch)"):
if not hasattr(instance, "branch_priorities"):
instance.branch_priorities = self.compute_priorities(instance)
x, y = self.x(training_instances), self.y(training_instances)