config.json

{
  "architectures": [
    "MoleculeSTMForGraphPrediction"
  ],
  "model_type": "molecule_stm",
  "initializer_range": 0.02,
  "num_layer": 5,
  "emb_dim": 300,
  "num_tasks": 1, 
  "graph_pooling": "mean" 
}