| { | |
| "model_type": "lstm", | |
| "vocab_size": 10000, | |
| "embedding_dim": 128, | |
| "hidden_size": 64, | |
| "num_layers": 2, | |
| "dropout_rate": 0.2, | |
| "max_sequence_length": 150, | |
| "batch_size": 8, | |
| "epochs": 50, | |
| "loss_function": "sparse_categorical_crossentropy", | |
| "optimizer": "adam", | |
| "metrics": [ | |
| "accuracy" | |
| ], | |
| "train_data_size": 5000, | |
| "categories": [ | |
| "Lifestyle", | |
| "Movies & Entertainment", | |
| "Food & Dining", | |
| "Others" | |
| ] | |
| } |