From 14c269130fe90c35b54eba6af253bc60edfa422d Mon Sep 17 00:00:00 2001 From: Mohammed Ajil Date: Thu, 23 Sep 2021 20:45:41 +0200 Subject: [PATCH 1/2] add support for quantile objective for LGBM models Signed-off-by: Mohammed Ajil --- onnxmltools/convert/lightgbm/operator_converters/LightGbm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxmltools/convert/lightgbm/operator_converters/LightGbm.py b/onnxmltools/convert/lightgbm/operator_converters/LightGbm.py index 423de707..415b63a3 100644 --- a/onnxmltools/convert/lightgbm/operator_converters/LightGbm.py +++ b/onnxmltools/convert/lightgbm/operator_converters/LightGbm.py @@ -444,7 +444,7 @@ def convert_lightgbm(scope, operator, container): elif gbm_text['objective'].startswith('multiclass'): n_classes = gbm_text['num_class'] attrs['post_transform'] = 'SOFTMAX' - elif gbm_text['objective'].startswith('regression'): + elif gbm_text['objective'].startswith(('regression', 'quantile')): n_classes = 1 # Regressor has only one output variable attrs['post_transform'] = 'NONE' attrs['n_targets'] = n_classes From 5e3545ee29d7ce1ec5a122f102b97bbae60d2767 Mon Sep 17 00:00:00 2001 From: Mohammed Ajil Date: Mon, 4 Oct 2021 08:34:36 +0200 Subject: [PATCH 2/2] add test case Signed-off-by: Mohammed Ajil --- tests/lightgbm/test_objective_functions.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/lightgbm/test_objective_functions.py b/tests/lightgbm/test_objective_functions.py index 9fc2db8a..b9bd2f91 100644 --- a/tests/lightgbm/test_objective_functions.py +++ b/tests/lightgbm/test_objective_functions.py @@ -31,6 +31,7 @@ class ObjectiveTest(unittest.TestCase): "regression", "poisson", "gamma", + "quantile" ) @staticmethod