| { | |
| "model_type": "lstm", | |
| "base_model": "LSTM", | |
| "library_name": "tensorflow", | |
| "vocab_size": 500, | |
| "embedding_dim": 64, | |
| "hidden_size": 64, | |
| "num_layers": 2, | |
| "dropout_rate": 0.2, | |
| "max_sequence_length": 10, | |
| "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" | |
| ] | |
| } | |