| { | |
| "version": 1, | |
| "classes": [ | |
| "glass", | |
| "metal", | |
| "organic", | |
| "paper", | |
| "plastic" | |
| ], | |
| "image_size": 224, | |
| "model_type": "baseline_cnn", | |
| "batch_size": 32, | |
| "seed": 123, | |
| "epochs_ran": 24, | |
| "best_val_accuracy": 0.6768447756767273, | |
| "best_val_loss": 0.8843007683753967, | |
| "test_loss": 0.8580112457275391, | |
| "test_accuracy": 0.6952140927314758, | |
| "test_precision_macro": 0.6960417768138908, | |
| "test_recall_macro": 0.6916533741850668, | |
| "test_f1_macro": 0.6924287856071965, | |
| "preprocessing": "PIL RGB resize to 224x224; model contains Rescaling(1/255)", | |
| "augmentation": [ | |
| "RandomFlip(horizontal)", | |
| "RandomRotation(0.1)", | |
| "RandomZoom(0.1)", | |
| "RandomContrast(0.1)" | |
| ] | |
| } |