JRJ/model-registry
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Support sklearn.tree.DecisionTreeClassifier

Browse Source
This commit is contained in:
Jason Jafari 2024-06-19 22:39:15 -04:00
parent 8e7f93ac03
commit d7a8ea9dc6
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
DOCS.md
View File

@ -299,6 +299,10 @@ supportedModels = {
"supported": True, "supported": True,
"normalPredictorFunction": "predict_proba" "normalPredictorFunction": "predict_proba"
}, },
"sklearn.tree.DecisionTreeClassifier": {
"supported": True,
"normalPredictorFunction": "predict_proba"
},
} }
``` ```

4
mlModelSaver/__init__.py
View File

@ -47,6 +47,10 @@ supportedModels = {
"supported": True, "supported": True,
"normalPredictorFunction": "predict_proba" "normalPredictorFunction": "predict_proba"
}, },
"sklearn.tree.DecisionTreeClassifier": {
"supported": True,
"normalPredictorFunction": "predict_proba"
},
} }
supportedDataType = { supportedDataType = {