Upload params.yaml
Browse files- params.yaml +16 -0
params.yaml
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MODEL_NAME: DenseNet121
|
| 2 |
+
IMAGE_SIZE: [224, 224, 3]
|
| 3 |
+
|
| 4 |
+
DATASET_NAME: TrashNet
|
| 5 |
+
SPLIT_RATIO: [0.9, 0.1]
|
| 6 |
+
LABEL_LIST: ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
|
| 7 |
+
|
| 8 |
+
SEED: 42
|
| 9 |
+
BATCH_SIZE: 16
|
| 10 |
+
EPOCHS: 25
|
| 11 |
+
LEARNING_RATE: 0.001
|
| 12 |
+
LOSS_FUNCTION: sparse_categorical_crossentropy
|
| 13 |
+
METRICS: ['accuracy']
|
| 14 |
+
|
| 15 |
+
NORMALIZE_CONFUSION_MATRIX: True
|
| 16 |
+
FIGSIZE: [18, 12]
|