From ea12e21e02e486f99eab1b5d8b47fddba9470b24 Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Tue, 23 Jun 2020 10:48:30 -0500 Subject: [PATCH] PrimalComponent: handle case where classifier is a fixed int --- src/python/miplearn/components/primal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/python/miplearn/components/primal.py b/src/python/miplearn/components/primal.py index d430768..a0d993e 100644 --- a/src/python/miplearn/components/primal.py +++ b/src/python/miplearn/components/primal.py @@ -90,7 +90,7 @@ class PrimalSolutionComponent(Component): if (category, label) not in self.classifiers.keys(): continue 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)]) else: ws = clf.predict_proba(x_test[category])