From 3149f55e39bd3b6b8d27c6aa212c892c4af25259 Mon Sep 17 00:00:00 2001 From: PyCaret Date: Tue, 14 Jul 2020 16:10:46 -0400 Subject: [PATCH] pycaret-nightly==0.25 --- pycaret/anomaly.py | 2 +- pycaret/classification.py | 13 ++++++++----- pycaret/clustering.py | 2 +- pycaret/regression.py | 8 ++++---- pycaret/utils.py | 2 +- setup.py | 2 +- 6 files changed, 16 insertions(+), 13 deletions(-) diff --git a/pycaret/anomaly.py b/pycaret/anomaly.py index 788b88a..eec595a 100644 --- a/pycaret/anomaly.py +++ b/pycaret/anomaly.py @@ -2,7 +2,7 @@ # Author: Moez Ali # License: MIT # Release: PyCaret 2.0x -# Last modified : 09/07/2020 +# Last modified : 14/07/2020 def setup(data, categorical_features = None, diff --git a/pycaret/classification.py b/pycaret/classification.py index 0f5d4c4..b8fc0ac 100644 --- a/pycaret/classification.py +++ b/pycaret/classification.py @@ -2,7 +2,7 @@ # Author: Moez Ali # License: MIT # Release: PyCaret 2.0x -# Last modified : 09/07/2020 +# Last modified : 14/07/2020 def setup(data, target, @@ -2297,7 +2297,10 @@ def create_model(estimator = None, if 'catboost' in mn: mn = 'CatBoostClassifier' - full_name = model_dict_logging.get(mn) + if mn in model_dict_logging.keys(): + full_name = model_dict_logging.get(mn) + else: + full_name = mn else: @@ -2306,10 +2309,10 @@ def create_model(estimator = None, if 'catboost' in mn: mn = 'CatBoostClassifier' - try: + if mn in model_dict_logging.keys(): full_name = model_dict_logging.get(mn) - except: - full_name = 'Custom Model' + else: + full_name = mn logger.info(str(full_name) + ' Imported succesfully') diff --git a/pycaret/clustering.py b/pycaret/clustering.py index 6f6d5dc..fba7d58 100644 --- a/pycaret/clustering.py +++ b/pycaret/clustering.py @@ -2,7 +2,7 @@ # Author: Moez Ali # License: MIT # Release: PyCaret 2.0x -# Last modified : 09/07/2020 +# Last modified : 14/07/2020 def setup(data, categorical_features = None, diff --git a/pycaret/regression.py b/pycaret/regression.py index b48ef2b..aada28c 100644 --- a/pycaret/regression.py +++ b/pycaret/regression.py @@ -2,7 +2,7 @@ # Author: Moez Ali # License: MIT # Release: PyCaret 2.0x -# Last modified : 09/07/2020 +# Last modified : 14/07/2020 def setup(data, target, @@ -2289,9 +2289,9 @@ def create_model(estimator = None, if 'catboost' in mn: mn = 'CatBoostRegressor' - full_name = model_dict_logging.get(mn) - - if full_name is None: + if mn in model_dict_logging.keys(): + full_name = model_dict_logging.get(mn) + else: full_name = mn logger.info(str(full_name) + ' Imported succesfully') diff --git a/pycaret/utils.py b/pycaret/utils.py index dad7d9d..1d43972 100644 --- a/pycaret/utils.py +++ b/pycaret/utils.py @@ -2,7 +2,7 @@ # Author: Moez Ali # License: MIT -version_ = "pycaret-nightly-0.24" +version_ = "pycaret-nightly-0.25" def version(): print(version_) diff --git a/setup.py b/setup.py index e8db567..fd18723 100644 --- a/setup.py +++ b/setup.py @@ -13,7 +13,7 @@ with open('requirements.txt') as f: setup( name="pycaret-nightly", - version="0.24", + version="0.25", description="Nightly build of PyCaret - An open source, low-code machine learning library in Python.", long_description=readme(), long_description_content_type="text/markdown", -- GitLab