Fix bug

Signed-off-by: xavier dupré <xavier.dupre@gmail.com>
onnx · Jul 27, 2021 · 0275cbc · 0275cbc
1 parent b998020
commit 0275cbc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/skl2onnx/operator_converters/support_vector_machines.py b/skl2onnx/operator_converters/support_vector_machines.py
@@ -250,10 +250,10 @@ def convert_sklearn_svm_classifier(
         #     transformed_confidences = (
         #         sum_of_confidences / (3 * (np.abs(sum_of_confidences) + 1)))
         #     return votes + transformed_confidences
-        if list(op.classes_) != list(range(len(op.classes_))):
-            raise RuntimeError(
-                "Classes different from first n integers are not supported "
-                "in SVC converter.")
+        # if list(op.classes_) != list(range(len(op.classes_))):
+        #     raise RuntimeError(
+        #         "Classes different from first n integers are not supported "
+        #         "in SVC converter.")
 
         cst3 = scope.get_unique_variable_name('cst3')
         container.add_initializer(cst3, proto_dtype, [], [3])