cla1r3 commited on
Commit
a5fab65
·
verified ·
1 Parent(s): 619a795

Create README

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