From 82b48c664ba0ace1b863e8935b754d398d2126f5 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Tue, 24 Oct 2023 22:30:28 -0700 Subject: [PATCH] Doc fix --- python/treelite/sklearn/importer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/treelite/sklearn/importer.py b/python/treelite/sklearn/importer.py index 7e33dd33..a114f7b8 100644 --- a/python/treelite/sklearn/importer.py +++ b/python/treelite/sklearn/importer.py @@ -101,8 +101,8 @@ def import_model(sklearn_model): import treelite.sklearn model = treelite.sklearn.import_model(clf) - Notes - ----- + Note + ---- This function does not yet support categorical splits in HistGradientBoostingRegressor and HistGradientBoostingClassifier. If you are using either estimator types, make sure that all test nodes have numerical test conditions.