diff --git a/.vscode/settings.json b/.vscode/settings.json index 16567b2..bfe30f7 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -1,6 +1,7 @@ { "cSpell.words": [ "Jafari", + "Logit", "statsmodels" ] } \ No newline at end of file diff --git a/mlModelSaver/__init__.py b/mlModelSaver/__init__.py index 2cb0aa6..876bcce 100644 --- a/mlModelSaver/__init__.py +++ b/mlModelSaver/__init__.py @@ -36,6 +36,9 @@ def check_file_exists(file_path): supportedModels = { "sm.OLS": { "supported": True + }, + "sm.Logit": { + "supported": True } } @@ -48,6 +51,9 @@ supportedDataType = { }, "binary":{ "supported": True + }, + "probebility":{ + "supported": True } } diff --git a/package.json b/package.json index 8d9316f..f1cef28 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "mlModelSaver", - "version": "1.0.16", + "version": "1.0.17", "description": "Make life easier for save and serving ml models", "main": "index.js", "repository": "git@github.com:smartdev-ca/mlModelSaver.git", diff --git a/setup.py b/setup.py index e1cfee7..6f67a4b 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name='mlModelSaver', - version='1.0.16', + version='1.0.17', packages=find_packages(), description='Make life easier for saving and serving ML models', long_description=open('DOCS.md').read(), # Assumes you have a README.md file