prediction-heads / README.md
jschlensok's picture
Update model card: add trizod2, document releases
c432c1c verified
|
Raw
History Blame Contribute Delete
1.22 kB
metadata
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, 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.