| { | |
| "model_type": "AttentiveFP", | |
| "task_type": "regression", | |
| "endpoint": "FBC", | |
| "hyperparameters": { | |
| "name": "baseline", | |
| "hidden_channels": 64, | |
| "num_layers": 2, | |
| "num_timesteps": 2, | |
| "dropout": 0.2, | |
| "learning_rate": 0.001, | |
| "weight_decay": 1e-05, | |
| "batch_size": 32, | |
| "epochs": 50, | |
| "patience": 10 | |
| }, | |
| "input_features": { | |
| "atom_features": 10, | |
| "bond_features": 6 | |
| }, | |
| "framework": "pytorch_geometric" | |
| } |