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

Support sklearn.tree.DecisionTreeRegressor

Browse Source
This commit is contained in:
Jason Jafari 2024-06-20 03:29:54 -04:00
parent a6ed3d96e2
commit 17f6a99fa2
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlModelSaver/__init__.py
View File

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

3
pytests/test_mlModelSaver.py
View File

@ -27,7 +27,8 @@ def test_ensureCLassInstance():
'sm.OLS', 'sm.OLS',
'sm.Logit', 'sm.Logit',
'sklearn.neighbors.KNeighborsClassifier', 'sklearn.neighbors.KNeighborsClassifier',
'sklearn.tree.DecisionTreeClassifier' 'sklearn.tree.DecisionTreeClassifier',
'sklearn.tree.DecisionTreeRegressor',
] ]