updated the config file.
Browse files- config.json +4 -7
config.json
CHANGED
|
@@ -3,8 +3,9 @@
|
|
| 3 |
"layers": [
|
| 4 |
{
|
| 5 |
"type": "embedding",
|
| 6 |
-
"
|
| 7 |
-
"
|
|
|
|
| 8 |
},
|
| 9 |
{
|
| 10 |
"type": "lstm",
|
|
@@ -32,9 +33,5 @@
|
|
| 32 |
}
|
| 33 |
],
|
| 34 |
"optimizer": "adam",
|
| 35 |
-
"loss": "mean_squared_error"
|
| 36 |
-
"metrics": ["accuracy"],
|
| 37 |
-
"learning_rate": 0.001,
|
| 38 |
-
"batch_size": 32,
|
| 39 |
-
"epochs": 10
|
| 40 |
}
|
|
|
|
| 3 |
"layers": [
|
| 4 |
{
|
| 5 |
"type": "embedding",
|
| 6 |
+
"input_dim": 97649,
|
| 7 |
+
"output_dim": 100,
|
| 8 |
+
"input_length": 100
|
| 9 |
},
|
| 10 |
{
|
| 11 |
"type": "lstm",
|
|
|
|
| 33 |
}
|
| 34 |
],
|
| 35 |
"optimizer": "adam",
|
| 36 |
+
"loss": "mean_squared_error"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
}
|