| { | |
| "model_type": "stability_predictor", | |
| "architecture": "esm2_mlp", | |
| "esm_model": "esm2_t6_8M_UR50D", | |
| "esm_params": 8000000, | |
| "embed_dim": 320, | |
| "repr_layer": 6, | |
| "freeze_esm": true, | |
| "head": { | |
| "hidden_dims": [256, 128], | |
| "dropout": 0.1, | |
| "activation": "relu" | |
| }, | |
| "training": { | |
| "dataset": "FLIP_meltome", | |
| "epochs": 16, | |
| "learning_rate": 0.001, | |
| "batch_size": 8, | |
| "early_stopping_patience": 15, | |
| "validation_r2": 0.616 | |
| }, | |
| "output": { | |
| "type": "regression", | |
| "description": "Normalized thermal stability score" | |
| } | |
| } | |