metadata
datasets:
- garythung/trashnet
metrics:
- accuracy
pipeline_tag: image-classification
Model Version 1.2
Created on: 2024-12-21 21:58:40
Metrics
- train_loss: 0.5705
- train_accuracy: 0.7812
- val_loss: 0.5975
- val_accuracy: 0.7752
Configuration
- architecture: Tensorflow CNN
- input_size: (224, 224, 3)
- num_classes: 6
- augmentation: {'rotation_range': 20, 'width_shift_range': 0.2, 'height_shift_range': 0.2, 'horizontal_flip': True, 'vertical_flip': True, 'shear_range': 0.2, 'brightness_range': [0.8, 1.2]}
- optimizer: adam
- learning_rate: 0.001
- batch_size: 32
- epochs: 50