whiteh4t commited on
Commit
b6f11ae
·
verified ·
1 Parent(s): c87881a

Standardize BGC-SetNet naming in model card

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -18,10 +18,10 @@ This repository contains the final five-seed model artifacts for the manuscript
18
 
19
  The release includes two model families:
20
 
21
- - **Pfam-augmented SetNet** (`pfam_setnet`): a Set Transformer encoder over frozen ESM-2 gene embeddings, relative gene positions, padding masks, and a BGC-level Pfam inventory embedding.
22
  - **Weighted Pfam Jaccard** (`weighted_pfam_jaccard`): learned non-negative Pfam-domain weights used in a weighted set-Jaccard retrieval baseline.
23
 
24
- The manuscript's main result is conservative: Pfam-domain content remains the strongest signal for this silver-label retrieval benchmark, while the ESM/Pfam SetNet ensemble gives small, statistically unsupported gains on some secondary metrics.
25
 
26
  ## Repository Contents
27
 
@@ -50,9 +50,9 @@ Family means over five optimization seeds and 16 held-out MIBiG-reference groups
50
  | Method | Recall@50 | MRR | MAP | nDCG@50 |
51
  |---|---:|---:|---:|---:|
52
  | Raw ESM mean | 0.7946 | 0.2550 | 0.7251 | 0.8078 |
53
- | Pfam-augmented SetNet | 0.8472 | 0.2786 | 0.7771 | 0.8502 |
54
  | Pfam Jaccard | 0.8788 | 0.3071 | 0.8480 | 0.9042 |
55
- | ESM/Pfam SetNet ensemble | 0.8769 | 0.3096 | 0.8503 | 0.9058 |
56
  | Weighted Pfam Jaccard | 0.8789 | 0.3069 | 0.8477 | 0.9040 |
57
 
58
  The weighted Pfam Recall@50 advantage over unweighted Pfam is numerically tiny (+0.00003) and should not be interpreted as a meaningful improvement.
 
18
 
19
  The release includes two model families:
20
 
21
+ - **Pfam-augmented BGC-SetNet** (`pfam_setnet`): a Set Transformer encoder over frozen ESM-2 gene embeddings, relative gene positions, padding masks, and a BGC-level Pfam inventory embedding.
22
  - **Weighted Pfam Jaccard** (`weighted_pfam_jaccard`): learned non-negative Pfam-domain weights used in a weighted set-Jaccard retrieval baseline.
23
 
24
+ The manuscript's main result is conservative: Pfam-domain content remains the strongest signal for this silver-label retrieval benchmark, while the ESM + BGC-SetNet + Pfam ensemble gives small, statistically unsupported gains on some secondary metrics.
25
 
26
  ## Repository Contents
27
 
 
50
  | Method | Recall@50 | MRR | MAP | nDCG@50 |
51
  |---|---:|---:|---:|---:|
52
  | Raw ESM mean | 0.7946 | 0.2550 | 0.7251 | 0.8078 |
53
+ | BGC-SetNet + Pfam | 0.8472 | 0.2786 | 0.7771 | 0.8502 |
54
  | Pfam Jaccard | 0.8788 | 0.3071 | 0.8480 | 0.9042 |
55
+ | ESM + BGC-SetNet + Pfam | 0.8769 | 0.3096 | 0.8503 | 0.9058 |
56
  | Weighted Pfam Jaccard | 0.8789 | 0.3069 | 0.8477 | 0.9040 |
57
 
58
  The weighted Pfam Recall@50 advantage over unweighted Pfam is numerically tiny (+0.00003) and should not be interpreted as a meaningful improvement.