Create config.json
Browse files- config.json +32 -0
config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "bidirectional_lstm",
|
| 3 |
+
"framework": "tensorflow",
|
| 4 |
+
"version": "1.0.0",
|
| 5 |
+
"architecture": {
|
| 6 |
+
"input_shape": [30, 13],
|
| 7 |
+
"output_shape": [1],
|
| 8 |
+
"layers": [
|
| 9 |
+
{"type": "bidirectional_lstm", "units": 64, "return_sequences": true},
|
| 10 |
+
{"type": "dropout", "rate": 0.2},
|
| 11 |
+
{"type": "bidirectional_lstm", "units": 32, "return_sequences": false},
|
| 12 |
+
{"type": "dropout", "rate": 0.2},
|
| 13 |
+
{"type": "dense", "units": 32, "activation": "relu"},
|
| 14 |
+
{"type": "dropout", "rate": 0.2},
|
| 15 |
+
{"type": "dense", "units": 1, "activation": "linear"}
|
| 16 |
+
]
|
| 17 |
+
},
|
| 18 |
+
"training": {
|
| 19 |
+
"optimizer": "adam",
|
| 20 |
+
"learning_rate": 0.0001,
|
| 21 |
+
"loss": "mse",
|
| 22 |
+
"batch_size": 32,
|
| 23 |
+
"epochs": 25,
|
| 24 |
+
"early_stopping_patience": 15,
|
| 25 |
+
"random_seed": 42
|
| 26 |
+
},
|
| 27 |
+
"performance": {
|
| 28 |
+
"test_rmse": 6.38,
|
| 29 |
+
"test_mae": 3.82,
|
| 30 |
+
"test_r2": 0.556
|
| 31 |
+
}
|
| 32 |
+
}
|