ryefoxlime commited on
Commit
c7308c1
1 Parent(s): 3c6a538

Pneumonia Detection with an accuracy of 91%

Browse files

Model Architecture:
- model = Sequential()
- model.add(base_model)
- model.add(AveragePooling2D(pool_size=(2, 2), strides=2))
- model.add(Flatten())
- model.add(Dense(256, activation="relu"))
- model.add(Dense(128, activation="relu"))
- model.add(Dense(2, activation="softmax"))
- model.compile(
loss="sparse_categorical_crossentropy",
optimizer=Adam(learning_rate=0.000035),
metrics=["sparse_categorical_accuracy"],
)

Confusion Matrix for the model

![image.png](https://cdn-uploads.huggingface.co/production/uploads/652917fd8297110ffe4e04ba/mS3PUyUzApUaP5NX2cgof.png)

model evaluation result:

![image.png](https://cdn-uploads.huggingface.co/production/uploads/652917fd8297110ffe4e04ba/ydOLZjDBtIKFiIvPE9IcF.png)

.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
fingerprint.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea0bc12205403b5a0f0f728a263169450ebac621fc5e6dd02d8a613ff7ab23b7
3
+ size 57
keras_metadata.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e78fd18384553e3ca23e8d2165c4ddefc8aed2b50c87097f003539176ce91ea
3
+ size 608693
saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfa84562f32088f3db4025a2731ed822c2c693a778f40f221803e4b00303138d
3
+ size 2871709
variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a12b9d6ea0d7e1d6595c37fc92c2164eb5a6845e8c13019079fa5ac6e6fc99f2
3
+ size 151445000
variables/variables.index ADDED
Binary file (17.2 kB). View file