Update README.md
Browse files
README.md
CHANGED
|
@@ -33,9 +33,21 @@ This confusion matrix gives a detailed view of the classification performance fo
|
|
| 33 |
---
|
| 34 |
|
| 35 |
## Model Details 🧠
|
| 36 |
-
- **Architecture**:
|
| 37 |
-
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
---
|
| 41 |
|
|
|
|
| 33 |
---
|
| 34 |
|
| 35 |
## Model Details 🧠
|
| 36 |
+
- **Architecture**: Architecture:
|
| 37 |
+
The model has a multi-input architecture with two branches:
|
| 38 |
+
Image input branch:
|
| 39 |
+
Input shape: (150, 150, 3)
|
| 40 |
+
Three Conv2D layers with ReLU activation and MaxPooling2D layers
|
| 41 |
+
Flattening layer at the end
|
| 42 |
+
Numerical input branch:
|
| 43 |
+
Input shape: (5,) (for 5 numerical features)
|
| 44 |
+
One Dense layer with 64 units and ReLU activation
|
| 45 |
+
The branches are then combined using concatenation, followed by:
|
| 46 |
+
Dense layer with 128 units and ReLU activation
|
| 47 |
+
Output Dense layer with 1 unit and sigmoid activation
|
| 48 |
+
|
| 49 |
+
- **Optimizer**: Adam optimizer.
|
| 50 |
+
- **Loss Function**: Binary crossentropy
|
| 51 |
|
| 52 |
---
|
| 53 |
|