| { | |
| "model_type": "custom", | |
| "architectures": ["LSTM"], | |
| "library_name": "tensorflow", | |
| "task_specific_params": { | |
| "text-classification": { | |
| "vocab_size": 500, | |
| "embedding_dim": 64, | |
| "hidden_size": 64, | |
| "num_layers": 2, | |
| "dropout_rate": 0.2, | |
| "max_sequence_length": 10 | |
| } | |
| }, | |
| "training_params": { | |
| "batch_size": 8, | |
| "epochs": 50, | |
| "loss_function": "sparse_categorical_crossentropy", | |
| "optimizer": "adam", | |
| "metrics": ["accuracy"] | |
| }, | |
| "train_data_size": 5000, | |
| "id2label": { | |
| "0": "Lifestyle", | |
| "1": "Movies & Entertainment", | |
| "2": "Food & Dining", | |
| "3": "Others" | |
| }, | |
| "label2id": { | |
| "Lifestyle": 0, | |
| "Movies & Entertainment": 1, | |
| "Food & Dining": 2, | |
| "Others": 3 | |
| } | |
| } |