prediction-heads / README.md
jschlensok's picture
Update model card: add trizod2, document releases
c432c1c verified
|
Raw
History Blame Contribute Delete
1.22 kB
---
license: mit
library_name: onnx
tags:
- protein
- intrinsic-disorder
- protein-disorder-prediction
- onnx
- udonpred
---
# UdonPred prediction heads (ONNX)
The exported ONNX prediction heads for [UdonPred](https://github.com/davidwagemann/udonpred),
one per disorder definition. Each head maps per-residue **ProstT5**
(`Rostlab/ProstT5_fp16`) embeddings of shape `(L, 1024)` to per-residue disorder
scores.
| File | Target |
| --- | --- |
| `trizod.onnx` | TriZOD |
| `trizod2.onnx` | TriZOD, retrained on the newer TriZOD set |
| `chezod.onnx` | CheZOD |
| `disprot.onnx` | DisProt |
| `softdis.onnx` | SoftDis |
| `pdbflex.onnx` | PDBFlex |
| `atlas.onnx` | ATLAS |
| `plddt.onnx` | pLDDT |
These heads do **not** contain the protein language model. Precompute ProstT5
embeddings first (see the repository's `udonpred-embed` command / EMBEDDINGS.md),
then run the heads with `udonpred-caid` / `udonpred-predict`.
## Releases
Consumers pin a tagged revision (e.g. `udonpred-caid ... --revision v0.2.0`).
| Tag | Change |
| --- | --- |
| `v0.1.0` | Initial 7 heads (trizod, chezod, disprot, softdis, pdbflex, atlas, plddt). |
| `v0.2.0` | Updated `trizod`; added the retrained `trizod2` head. |