Kaylah072001 commited on
Commit
1a1a353
·
verified ·
1 Parent(s): cf2e5d3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -3
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**: Fine-tuned **ResNet101**.
37
- - **Optimizer**: Adam with learning rate scheduling.
38
- - **Loss Function**: Cross-Entropy Loss.
 
 
 
 
 
 
 
 
 
 
 
 
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