| { | |
| "model_type": "fragnet", | |
| "n_classes": 1, | |
| "atom_features": 167, | |
| "frag_features": 167, | |
| "edge_features": 17, | |
| "num_layer": 4, | |
| "num_heads": 4, | |
| "drop_ratio": 0.15, | |
| "h1": 256, | |
| "h2": 256, | |
| "h3": 256, | |
| "h4": 256, | |
| "act": "celu", | |
| "emb_dim": 128, | |
| "fthead": "FTHead2", | |
| "task_type": "regression" | |
| } |