ryefoxlime commited on
Commit
4fdd31c
1 Parent(s): 0b4bf8c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "model_type": "custom_resnet50",
3
  "architectures": ["ResNet50V2"],
4
- "input_shape": [224, 224, 3],
5
  "num_classes": 2,
6
  "class_names": ["NORMAL", "PNEUMONIA"],
7
  "learning_rate": 0.000035,
 
1
  {
2
+ "model_type": "X-Ray_Classifiction",
3
  "architectures": ["ResNet50V2"],
4
+ "input_shape": [224, 224, 1],
5
  "num_classes": 2,
6
  "class_names": ["NORMAL", "PNEUMONIA"],
7
  "learning_rate": 0.000035,