assertion

harvitronix · Nov 20, 2017 · fa9492a · fa9492a
1 parent 952ab7c
commit fa9492a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/data.py b/data.py
@@ -100,6 +100,8 @@ def get_class_one_hot(self, class_str):
         # Now one-hot it.
         label_hot = to_categorical(label_encoded, len(self.classes))
 
+        assert len(label_hot) == len(self.classes)
+
         return label_hot
 
     def split_train_test(self):