Merge branch 'develop'

amaiya · Aug 12, 2020 · d6991a8 · d6991a8
2 parents 3d6e0c6 + d0d5d08
commit d6991a8
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,6 +6,18 @@ Most recent releases are shown at the top. Each release shows:
 - **Changed**: Additional parameters, changes to inputs or outputs, etc
 - **Fixed**: Bug fixes that don't change documented behaviour
 
+## 0.19.6 (2020-08-12)
+
+### New:
+- N/A
+
+### Changed
+-N/A
+
+### Fixed:
+- include metrics check in `is_classifier` function to support with non-standard loss functions
+
+
 ## 0.19.5 (2020-08-11)
 
 ### New:

diff --git a/ktrain/utils.py b/ktrain/utils.py
@@ -79,6 +79,10 @@ def is_classifier(model):
                  'sparse_categorical_crossentropy',
                  'binary_crossentropy']:
         is_classifier = True
+    else:
+        mlist = metrics_from_model(model)
+        if isinstance(mlist, (list, np.ndarray)) and 'accuracy' in mlist:
+            is_classifier = True
 
     # check for multilabel
     if loss == 'binary_crossentropy':

diff --git a/ktrain/version.py b/ktrain/version.py
@@ -1,2 +1,2 @@
 __all__ = ['__version__']
-__version__ = '0.19.5'
+__version__ = '0.19.6'