Tech: Fix Config to enable Inference API
#15
by
ai-venkat-r
- opened
- 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 |
+
}
|