Config
Config file to define all hyperparameters.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 | from tensorflow.keras.preprocessing.image import ImageDataGenerator
epoch = 10
batch_size = 32
datagen = ImageDataGenerator()
step_per_epoch = 100
target_size = (128, 128)
x_col = None
y_col = None # TODO: Remove? This is not used if a task is defined.
perform_validation = False
validation_split = 0.1 # The percentage of data set aside for the validation set
validation_steps = 1
datagen_valid = ImageDataGenerator()
kwargs = {}
|