diff --git a/DOCS.md b/DOCS.md index b6a201c..94d4dfb 100644 --- a/DOCS.md +++ b/DOCS.md @@ -299,6 +299,10 @@ supportedModels = { "supported": True, "normalPredictorFunction": "predict_proba" }, + "sklearn.tree.DecisionTreeClassifier": { + "supported": True, + "normalPredictorFunction": "predict_proba" + }, } ``` diff --git a/mlModelSaver/__init__.py b/mlModelSaver/__init__.py index 9d61467..bb06dcc 100644 --- a/mlModelSaver/__init__.py +++ b/mlModelSaver/__init__.py @@ -47,6 +47,10 @@ supportedModels = { "supported": True, "normalPredictorFunction": "predict_proba" }, + "sklearn.tree.DecisionTreeClassifier": { + "supported": True, + "normalPredictorFunction": "predict_proba" + }, } supportedDataType = {