config.json

{
  "model_type": "custom",
  "architecture": "sequential",
  "layers": [
    {
      "type": "embedding",
      "input_dim": 97649,
      "output_dim": 100,
      "input_length": 100
    },
    {
      "type": "lstm",
      "units": 64,
      "activation": "tanh"
    },
    {
      "type": "dense",
      "units": 32,
      "activation": "relu"
    },
    {
      "type": "dropout",
      "rate": 0.5
    },
    {
      "type": "dense",
      "units": 16,
      "activation": "relu"
    },
    {
      "type": "dense",
      "units": 100,
      "activation": "sigmoid"
    }
  ],
  "optimizer": "adam",
  "loss": "mean_squared_error"
}