okeowo1014 commited on
Commit
625209e
1 Parent(s): 0cfc9ed

Update train3.py

Browse files
Files changed (1) hide show
  1. train3.py +3 -3
train3.py CHANGED
@@ -4,9 +4,9 @@ from tensorflow.keras.applications import VGG16
4
  from tensorflow.keras.layers import Flatten, Dense
5
 
6
  # Define data paths (modify as needed)
7
- train_data_dir = 'tt'
8
- validation_data_dir = 'tt'
9
- test_data_dir = 'tt'
10
 
11
  # Set image dimensions (adjust if necessary)
12
  img_width, img_height = 224, 224 # VGG16 expects these dimensions
 
4
  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