Support sklearn.tree.DecisionTreeClassifier

2024-06-19 22:39:15 -04:00 · 2024-06-19 22:39:15 -04:00 · d7a8ea9dc6
commit d7a8ea9dc6
parent 8e7f93ac03
2 changed files with 8 additions and 0 deletions
--- a/DOCS.md
+++ b/DOCS.md
@ -299,6 +299,10 @@ supportedModels = {
        "supported": True,
        "normalPredictorFunction": "predict_proba"
    },
+    "sklearn.tree.DecisionTreeClassifier": {
+        "supported": True,
+        "normalPredictorFunction": "predict_proba"
+    },
 }
 ```

--- 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 = {