From 3bea5207b3e241d29e37b829acb3bf97f2dd151b Mon Sep 17 00:00:00 2001
From: yuvaram singh <yuvaramsingh94@gmail.com>
Date: Sun, 5 Apr 2020 01:54:34 +0530
Subject: [PATCH] Fixing ValueError

ValueError: Invalid class_mode: other; expected one of: {'raw', 'input', 'binary', 'categorical', None, 'multi_output', 'sparse'}


https://www.tensorflow.org/api_docs/python/tf/keras/preprocessing/image/ImageDataGenerator#flow_from_dataframe
---
 utils/loaders.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/loaders.py b/utils/loaders.py
index 27c4253397..9a36df3139 100644
--- a/utils/loaders.py
+++ b/utils/loaders.py
@@ -35,7 +35,7 @@ def build(self, att, batch_size, label = None):
                 , x_col='image_id'
                 , y_col=label
                 , target_size=self.target_size 
-                , class_mode='other'
+                , class_mode='raw'
                 , batch_size=batch_size
                 , shuffle=True
             )