Fixed gradient boosting classifier converter mismatch on binary datas…

…et (#230)
onnx · Aug 8, 2019 · 160200e · 160200e
1 parent d026f89
commit 160200e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/skl2onnx/operator_converters/gradient_boosting.py b/skl2onnx/operator_converters/gradient_boosting.py
@@ -35,8 +35,7 @@ def convert_sklearn_gradient_boosting_classifier(scope, operator, container):
         if op.n_classes_ == 2:
             # class_prior_ was introduced in scikit-learn 0.21.
             if hasattr(op.init_, 'class_prior_'):
-                base_values = op.init_.class_prior_
-                assert base_values.shape == (2, )
+                base_values = [op.init_.class_prior_[0]]
             else:
                 base_values = [op.init_.prior]
             if op.loss == 'deviance':