Venkat.R commited on
Commit
7d08383
1 Parent(s): 7ec1e86

tech: update config to enable inference

Browse files
Files changed (1) hide show
  1. config.json +7 -7
config.json CHANGED
@@ -1,18 +1,18 @@
1
  {
2
  "model_type": "lstm",
3
- "vocab_size": 10000,
4
- "embedding_dim": 128,
 
 
5
  "hidden_size": 64,
6
  "num_layers": 2,
7
  "dropout_rate": 0.2,
8
- "max_sequence_length": 150,
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
+ }