File size: 348 Bytes
8fb06db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
MODEL_NAME: DenseNet121
IMAGE_SIZE: [224, 224, 3]
DATASET_NAME: TrashNet
SPLIT_RATIO: [0.9, 0.1]
LABEL_LIST: ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
SEED: 42
BATCH_SIZE: 16
EPOCHS: 25
LEARNING_RATE: 0.001
LOSS_FUNCTION: sparse_categorical_crossentropy
METRICS: ['accuracy']
NORMALIZE_CONFUSION_MATRIX: True
FIGSIZE: [18, 12] |