Wakka2905 commited on
Commit
809b2a2
·
1 Parent(s): 20c73bc

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +21 -0
config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "custom_neural_network",
3
+ "input_size": 7,
4
+ "output_size": 3,
5
+ "layers": [
6
+ {
7
+ "input": 7,
8
+ "output": 4,
9
+ "activation_function": "relu"
10
+ },
11
+ {
12
+ "input": 4,
13
+ "output": 3
14
+ }
15
+ ],
16
+ "agent": {
17
+ "learning_rate": 0.001,
18
+ "loss_function": "MSELoss",
19
+ "optimizer": "Adam"
20
+ }
21
+ }