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

RelaxationComponent: Always use np arrays

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-07 12:29:43 -06:00
parent 4057a65506
commit 3f1aec7fad
2 changed files with 64 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
miplearn/components/steps/drop_redundant.py
View File

@@ -5,6 +5,7 @@
import logging
from copy import deepcopy
import numpy as np
from tqdm import tqdm
from miplearn import Component
@@ -101,6 +102,8 @@ class DropRedundantInequalitiesStep(Component):
constraints[category] = []
x[category] += [instance.get_constraint_features(cid)]
constraints[category] += [cid]
for category in x.keys():
x[category] = np.array(x[category])
if return_constraints:
return x, constraints
else:
@@ -131,7 +134,7 @@ class DropRedundantInequalitiesStep(Component):
if category not in self.classifiers:
continue
y[category] = []
# x_cat = np.array(x_cat)
x_cat = np.array(x_cat)
proba = self.classifiers[category].predict_proba(x_cat)
for i in range(len(proba)):
if proba[i][1] >= self.threshold: