Skip to content

Commit 6ac6a11

Browse files
authored
Fixes #321, branch == for lightgbm converter (#333)
1 parent 69cc79f commit 6ac6a11

File tree

1 file changed

+4
-0
lines changed
  • onnxmltools/convert/lightgbm/operator_converters

1 file changed

+4
-0
lines changed

onnxmltools/convert/lightgbm/operator_converters/LightGbm.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,10 @@ def _translate_split_criterion(criterion):
2525
return 'BRANCH_GTE'
2626
elif criterion == '>':
2727
return 'BRANCH_GT'
28+
elif criterion == '==':
29+
return 'BRANCH_EQ'
30+
elif criterion == '!=':
31+
return 'BRANCH_NEQ'
2832
else:
2933
raise ValueError('Unsupported splitting criterion: %s. Only <=, <, >=, and > are allowed.')
3034

0 commit comments

Comments
 (0)