okeowo1014 commited on
Commit
3561ca3
1 Parent(s): e4cdd46

Update train3.py

Browse files
Files changed (1) hide show
  1. train3.py +2 -2
train3.py CHANGED
@@ -5,8 +5,8 @@ from tensorflow.keras.layers import Flatten, Dense
5
 
6
  # Define data paths (modify as needed)
7
  train_data_dir = 'train'
8
- validation_data_dir = 'validation'
9
- test_data_dir = 'test'
10
 
11
  # Set image dimensions (adjust if necessary)
12
  img_width, img_height = 224, 224 # VGG16 expects these dimensions
 
5
 
6
  # Define data paths (modify as needed)
7
  train_data_dir = 'train'
8
+ validation_data_dir = 'valid'
9
+ test_data_dir = 'valid'
10
 
11
  # Set image dimensions (adjust if necessary)
12
  img_width, img_height = 224, 224 # VGG16 expects these dimensions