Medtty commited on
Commit
4df8e95
·
verified ·
1 Parent(s): 35d2a59

Create model_config.json

Browse files
Files changed (1) hide show
  1. model_config.json +46 -0
model_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "neural_network",
3
+ "architecture": {
4
+ "layers": [
5
+ {
6
+ "type": "input",
7
+ "shape": [None, 784]
8
+ },
9
+ {
10
+ "type": "dense",
11
+ "units": 128,
12
+ "activation": "relu"
13
+ },
14
+ {
15
+ "type": "dense",
16
+ "units": 64,
17
+ "activation": "relu"
18
+ },
19
+ {
20
+ "type": "dense",
21
+ "units": 10,
22
+ "activation": "softmax"
23
+ }
24
+ ]
25
+ },
26
+ "training_config": {
27
+ "optimizer": "adam",
28
+ "loss": "categorical_crossentropy",
29
+ "metrics": ["accuracy"]
30
+ },
31
+ "hyperparameters": {
32
+ "learning_rate": 0.001,
33
+ "batch_size": 32,
34
+ "epochs": 50
35
+ },
36
+ "input_preprocessing": {
37
+ "normalization": "min_max",
38
+ "resize": [28, 28],
39
+ "color_mode": "grayscale"
40
+ },
41
+ "metadata": {
42
+ "created_at": "2024-01-01",
43
+ "version": "1.0.0",
44
+ "task": "classification"
45
+ }
46
+ }