fix converters for old versions of xgboost

Signed-off-by: Xavier Dupre <[email protected]>
xadupre · Dec 15, 2023 · 6edd641 · 6edd641
1 parent 21bb426
commit 6edd641
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/onnxmltools/convert/xgboost/shape_calculators/Classifier.py b/onnxmltools/convert/xgboost/shape_calculators/Classifier.py
@@ -38,8 +38,10 @@ def calculate_xgboost_classifier_output_shapes(operator):
         if objective == "reg:logistic" and ncl == 1:
             ncl = 2
     classes = xgb_node.classes_
-    if np.issubdtype(classes.dtype, np.floating) or np.issubdtype(
-        classes.dtype, np.integer
+    if (
+        np.issubdtype(classes.dtype, np.floating)
+        or np.issubdtype(classes.dtype, np.integer)
+        or np.issubdtype(classes.dtype, np.bool_)
     ):
         operator.outputs[0].type = Int64TensorType(shape=[N])
     else: