nielsr HF Staff commited on
Commit
00e8296
·
verified ·
1 Parent(s): 307e378

Update pipeline tag and add paper/code links

Browse files

Hi! I'm Niels from the community science team at Hugging Face.

This PR improves your model card by:
- Updating the `pipeline_tag` to `graph-ml` to better reflect the model's domain in protein structure and geometric deep learning.
- Adding a link to the paper: [Protein Fold Classification at Scale: Benchmarking and Pretraining](https://huggingface.co/papers/2605.18552).
- Adding a link to the official [GitHub repository](https://github.com/BorgwardtLab/TEDBench).

These changes help users discover the model more easily and provide direct access to the associated research and code.

Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -1,26 +1,26 @@
1
  ---
2
  library_name: tedbench
3
- tags:
4
- - protein
5
- - structure
6
- - fold-classification
7
- - tedbench
8
- pipeline_tag: feature-extraction
9
  license: bsd-3-clause
 
 
 
 
 
 
10
  ---
11
 
12
  # TEDBench — Pretrained autoencoder (structure + sequence)
13
 
14
  **Variant:** `miae_b` + seq  |  **Parameters:** 102M  |  **Layers:** 12  |  **Hidden dim:** 768  |  **Attn heads:** 12
15
 
16
- This is a **pretrained MiAE** checkpoint. Use it as a feature extractor or as the starting point for fine-tuning.
 
 
 
17
 
18
  > **+seq variant** — sequence embeddings are concatenated to the geometric encoder input (`model.use_seq_input=true`).
19
 
20
- Part of the [TEDBench](https://github.com/BorgwardtLab/TEDBench) benchmark for
21
- protein fold classification (ICML 2026). MiAE is an SE(3)-invariant masked
22
- autoencoder that masks up to 90% of backbone frames and reconstructs the full
23
- structure with a lightweight decoder.
24
 
25
  ## Architecture sizes
26
 
@@ -61,4 +61,4 @@ model.eval()
61
  booktitle={Proceedings of the 43rd International Conference on Machine Learning},
62
  year={2026}
63
  }
64
- ```
 
1
  ---
2
  library_name: tedbench
 
 
 
 
 
 
3
  license: bsd-3-clause
4
+ pipeline_tag: graph-ml
5
+ tags:
6
+ - protein
7
+ - structure
8
+ - fold-classification
9
+ - tedbench
10
  ---
11
 
12
  # TEDBench — Pretrained autoencoder (structure + sequence)
13
 
14
  **Variant:** `miae_b` + seq  |  **Parameters:** 102M  |  **Layers:** 12  |  **Hidden dim:** 768  |  **Attn heads:** 12
15
 
16
+ This repository contains the **pretrained MiAE** checkpoint introduced in the paper [Protein Fold Classification at Scale: Benchmarking and Pretraining](https://huggingface.co/papers/2605.18552). Use it as a feature extractor or as the starting point for fine-tuning.
17
+
18
+ - **Paper:** [Protein Fold Classification at Scale: Benchmarking and Pretraining](https://huggingface.co/papers/2605.18552)
19
+ - **Repository:** [BorgwardtLab/TEDBench](https://github.com/BorgwardtLab/TEDBench)
20
 
21
  > **+seq variant** — sequence embeddings are concatenated to the geometric encoder input (`model.use_seq_input=true`).
22
 
23
+ Part of the [TEDBench](https://github.com/BorgwardtLab/TEDBench) benchmark for protein fold classification (ICML 2026). MiAE is an SE(3)-invariant masked autoencoder that masks up to 90% of backbone frames and reconstructs the full structure with a lightweight decoder.
 
 
 
24
 
25
  ## Architecture sizes
26
 
 
61
  booktitle={Proceedings of the 43rd International Conference on Machine Learning},
62
  year={2026}
63
  }
64
+ ```