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.