steveGomez commited on
Commit
d6fc232
·
verified ·
1 Parent(s): 8348ccb

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +46 -0
config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "name": "Retinal Disease Classification Model",
4
+ "type": "Image to Text",
5
+ "input_shape": [224, 224, 3],
6
+ "output_shape": [1, "text"],
7
+ "layers": [
8
+ {
9
+ "type": "Conv2D",
10
+ "filters": 32,
11
+ "kernel_size": [3, 3],
12
+ "activation": "relu"
13
+ },
14
+ {
15
+ "type": "MaxPooling2D",
16
+ "pool_size": [2, 2]
17
+ },
18
+ {
19
+ "type": "Conv2D",
20
+ "filters": 64,
21
+ "kernel_size": [3, 3],
22
+ "activation": "relu"
23
+ },
24
+ {
25
+ "type": "MaxPooling2D",
26
+ "pool_size": [2, 2]
27
+ },
28
+ {
29
+ "type": "Flatten"
30
+ },
31
+ {
32
+ "type": "Dense",
33
+ "units": 128,
34
+ "activation": "relu"
35
+ },
36
+ {
37
+ "type": "Dense",
38
+ "units": 1,
39
+ "activation": "sigmoid"
40
+ }
41
+ ],
42
+ "loss": "binary_crossentropy",
43
+ "optimizer": "adam",
44
+ "metrics": ["accuracy"]
45
+ }
46
+ }