{
  "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
  }
}