File size: 543 Bytes
a5fab65 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | # Antibody Liability Predictor
This repository contains a PyTorch neural network for predicting antibody
developability liabilities from ESM-2 embeddings.
## Model architecture
- Input: 640-D (VH 320 + VL 320)
- Output: 4 regression values
- Architecture: MLP (128 → 64)
## Usage
```python
import torch
from model import LiabilityPredictor
model = LiabilityPredictor(input_dim=640)
model.load_state_dict(torch.load("liability_predictor.pt", map_location="cpu"))
model.eval()
# x should be shape (640,) or (batch, 640)
y_pred = model(x)
|