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

Implement tests for ConvertTightIneqsIntoEqsStep

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-07 10:29:22 -06:00
parent 0377b5b546
commit d8dc8471aa
4 changed files with 40 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
miplearn/components/steps/convert_tight.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
@@ -124,7 +125,7 @@ class ConvertTightIneqsIntoEqsStep(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: