{
  "categorical_dim": 31,
  "embedding_dim": 768,
  "hidden_dim": 512,
  "output_dim": 128
}