arittrabag commited on
Commit
72bda3b
·
verified ·
1 Parent(s): edce0f6

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +16 -0
config.json CHANGED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_dir": "Data",
3
+ "classes": [
4
+ "Mild Dementia",
5
+ "Moderate Dementia",
6
+ "Non Demented",
7
+ "Very Mild Dementia"
8
+ ],
9
+ "img_height": 224,
10
+ "img_width": 224,
11
+ "batch_size": 32,
12
+ "epochs": 10,
13
+ "learning_rate": 0.001,
14
+ "model_save_path": "alzheimers_detection_model.h5",
15
+ "test_image_path": "Data/Very mild Dementia/OAS1_0003_MR1_mpr-1_116.jpg"
16
+ }