Venkat.R
commited on
Commit
•
7d08383
1
Parent(s):
7ec1e86
tech: update config to enable inference
Browse files- config.json +7 -7
config.json
CHANGED
@@ -1,18 +1,18 @@
|
|
1 |
{
|
2 |
"model_type": "lstm",
|
3 |
-
"
|
4 |
-
"
|
|
|
|
|
5 |
"hidden_size": 64,
|
6 |
"num_layers": 2,
|
7 |
"dropout_rate": 0.2,
|
8 |
-
"max_sequence_length":
|
9 |
"batch_size": 8,
|
10 |
"epochs": 50,
|
11 |
"loss_function": "sparse_categorical_crossentropy",
|
12 |
"optimizer": "adam",
|
13 |
-
"metrics": [
|
14 |
-
"accuracy"
|
15 |
-
],
|
16 |
"train_data_size": 5000,
|
17 |
"categories": [
|
18 |
"Lifestyle",
|
@@ -20,4 +20,4 @@
|
|
20 |
"Food & Dining",
|
21 |
"Others"
|
22 |
]
|
23 |
-
}
|
|
|
1 |
{
|
2 |
"model_type": "lstm",
|
3 |
+
"base_model": "LSTM",
|
4 |
+
"library_name": "tensorflow",
|
5 |
+
"vocab_size": 500,
|
6 |
+
"embedding_dim": 64,
|
7 |
"hidden_size": 64,
|
8 |
"num_layers": 2,
|
9 |
"dropout_rate": 0.2,
|
10 |
+
"max_sequence_length": 10,
|
11 |
"batch_size": 8,
|
12 |
"epochs": 50,
|
13 |
"loss_function": "sparse_categorical_crossentropy",
|
14 |
"optimizer": "adam",
|
15 |
+
"metrics": ["accuracy"],
|
|
|
|
|
16 |
"train_data_size": 5000,
|
17 |
"categories": [
|
18 |
"Lifestyle",
|
|
|
20 |
"Food & Dining",
|
21 |
"Others"
|
22 |
]
|
23 |
+
}
|