Merge branch 'develop'

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.2 (2020-08-03)
+
+### New:
+- N/A
+
+### Changed
+- N/A
+
+### Fixed:
+- added missing `num_classes` argument to `to_categorical`
+
+
 ## 0.19.1 (2020-07-29)
 
 ### New:

ktrain/text/preprocessor.py

@@ -529,10 +529,8 @@ def _transform_y(self, y_data, verbose=1):
 
         # if shape is 1, this is either a classification or regression task 
         # depending on class_names existing
-        y_data = to_categorical(y_data) if len(y_data.shape) == 1 and self.get_classes() else y_data
+        y_data = to_categorical(y_data, num_classes=len(self.get_classes())) if len(y_data.shape) == 1 and self.get_classes() else y_data
         if self.get_classes():
-            x = y_data.shape[1]
-            y = len(self.get_classes())
             if train and y_data.shape[1] != len(self.get_classes()):
                 raise Exception('Class labels in training set are %s, but y_data has %s classes' % (self.get_classes(), y_data.shape[1]))
         return y_data

ktrain/version.py

@@ -1,2 +1,2 @@
 __all__ = ['__version__']
-__version__ = '0.19.1'
+__version__ = '0.19.2'