use sparse_categorical_crossentropy

J13253902839 · Aug 18, 2020 · 98460fd · 98460fd
1 parent bd90265
commit 98460fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/generator.py b/src/generator.py
@@ -26,8 +26,8 @@ def __getitem__(self, idx):
             ages.append(row["ages"])
 
         imgs = np.asarray(imgs)
-        genders = to_categorical(genders, 2)
-        ages = to_categorical(ages, 101)
+        genders = np.asarray(genders)
+        ages = np.asarray(ages)
 
         return imgs, (genders, ages)