pycaret-nightly==0.25

pycaret/anomaly.py

@@ -2,7 +2,7 @@
 # Author: Moez Ali <moez.ali@queensu.ca>
 # License: MIT
 # Release: PyCaret 2.0x
-# Last modified : 09/07/2020
+# Last modified : 14/07/2020
 
 def setup(data, 
           categorical_features = None,

pycaret/classification.py

@@ -2,7 +2,7 @@
 # Author: Moez Ali <moez.ali@queensu.ca>
 # 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')
 

pycaret/clustering.py

@@ -2,7 +2,7 @@
 # Author: Moez Ali <moez.ali@queensu.ca>
 # License: MIT
 # Release: PyCaret 2.0x
-# Last modified : 09/07/2020
+# Last modified : 14/07/2020
 
 def setup(data, 
         categorical_features = None,

pycaret/regression.py

@@ -2,7 +2,7 @@
 # Author: Moez Ali <moez.ali@queensu.ca>
 # 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')

pycaret/utils.py

@@ -2,7 +2,7 @@
 # Author: Moez Ali <moez.ali@queensu.ca>
 # License: MIT
 
-version_ = "pycaret-nightly-0.24"
+version_ = "pycaret-nightly-0.25"
 
 def version():
     print(version_)

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",