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

PrimalComponent: handle case where classifier is a fixed int

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-06-23 10:48:30 -05:00
parent 11d22b4c3c
commit ea12e21e02
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

@@ -90,7 +90,7 @@ class PrimalSolutionComponent(Component):
if (category, label) not in self.classifiers.keys(): if (category, label) not in self.classifiers.keys():
continue continue
clf = self.classifiers[category, label] clf = self.classifiers[category, label]
if isinstance(clf, float): if isinstance(clf, float) or isinstance(clf, int):
ws = np.array([[1 - clf, clf] for _ in range(n)]) ws = np.array([[1 - clf, clf] for _ in range(n)])
else: else:
ws = clf.predict_proba(x_test[category]) ws = clf.predict_proba(x_test[category])