From d7a8ea9dc64464c9c9cab0c99d0f0d9edce1496e Mon Sep 17 00:00:00 2001
From: Jason Jafari <me@jasonjafari.com>
Date: Wed, 19 Jun 2024 22:39:15 -0400
Subject: [PATCH] Support sklearn.tree.DecisionTreeClassifier

---
 DOCS.md                  | 4 ++++
 mlModelSaver/__init__.py | 4 ++++
 2 files changed, 8 insertions(+)

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